File tree Expand file tree Collapse file tree 2 files changed +15
-2
lines changed Expand file tree Collapse file tree 2 files changed +15
-2
lines changed Original file line number Diff line number Diff line change 1
1
diff --git a/aten/src/ATen/CMakeLists.txt b/aten/src/ATen/CMakeLists.txt
2
2
index f3281d3cb..c8b6ed30e 100644
3
- --- a/aten/src/ATen/CMakeLists.txt
3
+ --- a/aten/src/ATen/CMakeLists.txt
4
4
+++ b/aten/src/ATen/CMakeLists.txt
5
5
@@ -418,7 +418,7 @@ if(NOT MSVC AND NOT EMSCRIPTEN AND NOT INTERN_BUILD_MOBILE)
6
6
list(APPEND ATen_THIRD_PARTY_INCLUDE ${CMAKE_BINARY_DIR}/include)
@@ -20,3 +20,16 @@ index f3281d3cb..c8b6ed30e 100644
20
20
21
21
set(CMAKE_C_FLAGS_DEBUG ${OLD_CMAKE_C_FLAGS_DEBUG})
22
22
set(CMAKE_CXX_FLAGS ${OLD_CMAKE_CXX_FLAGS})
23
+ diff --git a/aten/src/ATen/cpu/vec/vec256/vec256_bfloat16.h b/aten/src/ATen/cpu/vec/vec256/vec256_bfloat16.h
24
+ index f3281d3cb..c8b6ed30e 100644
25
+ --- a/aten/src/ATen/cpu/vec/vec256/vec256_bfloat16.h
26
+ +++ b/aten/src/ATen/cpu/vec/vec256/vec256_bfloat16.h
27
+ @@ -266,7 +266,7 @@ public:
28
+ }
29
+ return b;
30
+ }
31
+ - Vectorized<T> map(const __m256 (*const vop)(__m256)) const {
32
+ + Vectorized<T> map(__m256 (*const vop)(__m256)) const {
33
+ __m256 lo, hi;
34
+ cvt_to_fp32<T>(values, lo, hi);
35
+ const auto o1 = vop(lo);
Original file line number Diff line number Diff line change 1
1
{
2
2
"versions" : [
3
3
{
4
- "git-tree" : " d6c80d406232a0bfd3360a40132a69d9f70b972c " ,
4
+ "git-tree" : " 2590eb541f5f4f5a959ae3b6990488b4b7d2c473 " ,
5
5
"version" : " 2.1.2" ,
6
6
"port-version" : 8
7
7
},
You can’t perform that action at this time.
0 commit comments