From 5682e6d34645db87d6656a6a1f90f8425cb0420e Mon Sep 17 00:00:00 2001 From: Jiarui Fang Date: Mon, 26 Dec 2022 16:45:14 +0800 Subject: [PATCH] [hotfix] correcnt cpu_optim runtime compilation (#2197) --- colossalai/kernel/op_builder/cpu_adam.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/colossalai/kernel/op_builder/cpu_adam.py b/colossalai/kernel/op_builder/cpu_adam.py index 63b16d9fd..136f604f2 100644 --- a/colossalai/kernel/op_builder/cpu_adam.py +++ b/colossalai/kernel/op_builder/cpu_adam.py @@ -48,7 +48,7 @@ class CPUAdamBuilder(Builder): sources=[os.path.join('colossalai/kernel/cuda_native/csrc', path) for path in self.sources], include_dirs=self.extra_include_paths, extra_compile_args={ - 'cxx': ['-O3'] + self.version_dependent_macros + self.extra_cuda_flags, + 'cxx': ['-O3'] + self.version_dependent_macros + self.extra_cxx_flags, 'nvcc': append_nvcc_threads(['-O3', '--use_fast_math'] + self.version_dependent_macros + self.extra_cuda_flags)