1 #include "caffe2/core/types.h" 2 #include "caffe2/perfkernels/cvtsh_ss_bugfix.h" 3 #include "caffe2/perfkernels/typed_axpy.h" 4 #include "caffe2/utils/math.h" 11 void TypedAxpy_float16_float__avx2_fma(
18 while (((
unsigned long)x % 16) && N) {
19 *(y++) += _cvtsh_ss((*(x++)).x) * a;
25 __m256 mma = _mm256_set1_ps(a);
27 const int bound = (N % 8) ? N - 8 : N;
29 for (; current < bound; current += 8) {
31 _mm_loadu_si128(reinterpret_cast<const __m128i*>(x + current));
32 __m256 mmx_32 = _mm256_cvtph_ps(mmx_16);
33 __m256 mmy = _mm256_loadu_ps(y + current);
34 mmy = _mm256_fmadd_ps(mmx_32, mma, mmy);
35 _mm256_storeu_ps(y + current, mmy);
40 y[current] += _cvtsh_ss(x[current].x) * a;
46 void TypedAxpy_uint8_float__avx2_fma(
49 const std::uint8_t* x,
53 while (((
unsigned long)x % 16) && N) {
54 *(y++) += (
float)(*(x++)) * a;
60 __m256 mma = _mm256_set1_ps(a);
62 const int bound = (N % 8) ? N - 8 : N;
64 for (; current < bound; current += 8) {
65 __m256i mmx_int32 = _mm256_cvtepi8_epi32(
66 _mm_loadu_si128(reinterpret_cast<const __m128i*>(x + current)));
67 __m256 mmx_fp32 = _mm256_cvtepi32_ps(mmx_int32);
69 __m256 mmy = _mm256_loadu_ps(y + current);
70 mmy = _mm256_fmadd_ps(mmx_fp32, mma, mmy);
71 _mm256_storeu_ps(y + current, mmy);
76 y[current] += (float)(x[current]) * a;
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...