aboutsummaryrefslogtreecommitdiff
path: root/misc/pytorch/files/patch-aten_src_ATen_cpu_vec_vec256_vec256__bfloat16.h
blob: e03ac51d837b44b203bf29d710122f4a28145f0d (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
--- aten/src/ATen/cpu/vec/vec256/vec256_bfloat16.h.orig	2024-03-27 22:28:51 UTC
+++ aten/src/ATen/cpu/vec/vec256/vec256_bfloat16.h
@@ -265,7 +265,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);