[hotfix] correcnt cpu_optim runtime compilation (hpcaitech#2197)

widyhu2020 · Dec 26, 2022 · 5682e6d · 5682e6d
1 parent 6642ceb
commit 5682e6d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/colossalai/kernel/op_builder/cpu_adam.py b/colossalai/kernel/op_builder/cpu_adam.py
@@ -48,7 +48,7 @@ def builder(self, name) -> 'CUDAExtension':
  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)