p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix avx512 compile option

This commit is contained in:
zhaosida 2021-11-27 15:40:35 +08:00
parent 6ab8ddfd3a
commit 7de4c114b3
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/CMakeLists.txt
View File

@ -130,7 +130,7 @@ if(ENABLE_CPU)
elseif("${X86_64_SIMD}" STREQUAL "avx")
target_compile_definitions(nnacl_mid PRIVATE ENABLE_SSE ENABLE_AVX)
elseif("${X86_64_SIMD}" STREQUAL "avx512")
target_compile_definitions(nnacl_mid PRIVATE ENABLE_AVX512)
target_compile_definitions(nnacl_mid PRIVATE ENABLE_SSE ENABLE_AVX ENABLE_AVX512)
target_compile_options(nnacl_mid PRIVATE -mavx512f)
endif()
target_compile_options(nnacl_mid PRIVATE -fPIC)

2
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/intrinsics/avx/TiledC8MatMulFp32.c
View File

@ -20,7 +20,6 @@
#include <x86intrin.h>
#endif
#include "nnacl/fp32/common_func_fp32.h"
#endif
void TiledC8MatmulFp32(float *dst, const float *src, const float *weight, size_t cal_num, size_t ic8, size_t oc8) {
const float *src_tmp = src;
@ -242,3 +241,4 @@ void TiledC8MatmulFp32(float *dst, const float *src, const float *weight, size_t
dst += cal_num;
}
}
#endif