From 1903a1380ef6d5a45f77c59002edf1b7120d0d05 Mon Sep 17 00:00:00 2001 From: Reza Yazdani Date: Fri, 26 Feb 2021 04:40:55 +0000 Subject: [PATCH] fixing the compiling issue for the AMD architecture --- op_builder/cpu_adam.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/op_builder/cpu_adam.py b/op_builder/cpu_adam.py index 20f8fe2d8b6f..054654d00c0f 100644 --- a/op_builder/cpu_adam.py +++ b/op_builder/cpu_adam.py @@ -36,7 +36,7 @@ def simd_width(self): return '-D__AVX512__' elif 'avx2' in result: return '-D__AVX256__' - return '' + return '-D__SCALAR__' def cxx_args(self): CUDA_LIB64 = os.path.join(torch.utils.cpp_extension.CUDA_HOME, "lib64")