Skip to content

GitLab

Unverified Commit 490e6f7c authored by Reza Yazdani's avatar Reza Yazdani Committed by GitHub
Browse files

fixing the compiling issue for the AMD architecture (#796) 


Co-authored-by: default avatarJeff Rasley <jerasley@microsoft.com>
parent 62396b71
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
op_builder/cpu_adam.py
View file @ 490e6f7c
...@@ -36,7 +36,7 @@ class CPUAdamBuilder(CUDAOpBuilder): ...@@ -36,7 +36,7 @@ class CPUAdamBuilder(CUDAOpBuilder):
return '-D__AVX512__' return '-D__AVX512__'
elif 'avx2' in result: elif 'avx2' in result:
return '-D__AVX256__' return '-D__AVX256__'
return '' return '-D__SCALAR__'
def cxx_args(self): def cxx_args(self):
CUDA_LIB64 = os.path.join(torch.utils.cpp_extension.CUDA_HOME, "lib64") CUDA_LIB64 = os.path.join(torch.utils.cpp_extension.CUDA_HOME, "lib64")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment