[hotfix] correcnt cpu_optim runtime compilation (#2197)

2 years ago · 5682e6d346
parent 6642cebdbe
commit 5682e6d346
1 changed files with 1 additions and 1 deletions
--- 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)