3

我有这样的 AVX C++ 代码,在 Visual Studio 2010 下编译得很好:

#include <immintrin.h>
#include <iostream>

int main() {
    float data[] = {0, 1, 2, 3, 4, 5, 6, 7};
    __m256 ymm0 = _mm256_loadu_ps(data);
    // ..
    float r0 = ymm0.m256_f32[0];
    float r4 = ymm0.m256_f32[4];
    std::cout << r0 << " " << r4 << std::endl;
}

然而,GCC 给出了以下错误:

foo.cpp:8:18: error: request for member ‘m256_f32’ in ‘ymm0’, which is of non-class type ‘__m256 {aka __vector(8) float}’
foo.cpp:9:18: error: request for member ‘m256_f32’ in ‘ymm0’, which is of non-class type ‘__m256 {aka __vector(8) float}’

我做了一些研究,这似乎ymm0.m256_f32是 Microsoft 特定的指令,用于从长 AVX 寄存器中提取单个浮点数。但是我可以用什么来让 gcc/linux 做同样的事情呢?

4

2 回答 2

4

GCC 可以在 C 语言中索引向量,但不能在 C++ 中。您可以考虑将一小部分代码重写为 C。

另一种选择是显式使用随机播放、提取和转换内在函数 - _mm256_shuffle_pd, _mm256_extractf128_pd, _mm_cvtsd_f64.

于 2012-11-28T11:52:51.417 回答
1

在@chill 之后,我针对这个特定问题提出了以下两种解决方案。C++ 解决方案:

#include <immintrin.h>
#include <iostream>
int main() {
    float data[] = {0, 1, 2, 3, 4, 5, 6, 7};
    __m256 ymm0 = _mm256_loadu_ps(data);
    // ..
    __m128 low = _mm256_extractf128_ps (ymm0, 0); // extracts lower half
    __m128 high = _mm256_extractf128_ps (ymm0, 1);// extracts upper half
    float r0 = _mm_cvtss_f32 (low);  // extracts lowest float
    float r4 = _mm_cvtss_f32 (high);  // extracts lowest float
    std::cout << r0 << " " << r4 << std::endl;
}

以及纯 C 中的解决方案:

#include <immintrin.h>
#include <stdio.h>

int main() {
    float data[] = {0, 1, 2, 3, 4, 5, 6, 7};
    __m256 ymm0 = _mm256_loadu_ps(data);
    // ..
    float r0 = ymm0[0];
    float r4 = ymm0[4];
    printf("%f, %f", r0, r4);
}
于 2012-11-30T09:25:18.367 回答