misc/pytorch/files/patch-aten_src_ATen_cpu_vec_vec256_vec256__bfloat16.h


1
2
3
4
5
6
7
8
9
10
11

--- aten/src/ATen/cpu/vec/vec256/vec256_bfloat16.h.orig	2023-10-12 12:49:42 UTC
+++ aten/src/ATen/cpu/vec/vec256/vec256_bfloat16.h
@@ -266,7 +266,7 @@ static_assert( (public)
     }
     return b;
   }
-  Vectorized<T> map(const __m256 (*const vop)(__m256)) const {
+  Vectorized<T> map(__m256 (*const vop)(__m256)) const {
     __m256 lo, hi;
     cvt_to_fp32<T>(values, lo, hi);
     const auto o1 = vop(lo);