Unverified Commit 66c55151 authored by bdf's avatar bdf Committed by GitHub
Browse files

[Enhance] Add a default value for MMCV_MLU_ARGS (#2688)



* add mlu_args

* add mlu_args

* Modify the code

---------
Co-authored-by: default avatarbudefei <budefei@cambricon.com>
parent 4c4ba6cb
...@@ -369,7 +369,7 @@ def get_extensions(): ...@@ -369,7 +369,7 @@ def get_extensions():
exit() exit()
define_macros += [('MMCV_WITH_MLU', None)] define_macros += [('MMCV_WITH_MLU', None)]
mlu_args = os.getenv('MMCV_MLU_ARGS') mlu_args = os.getenv('MMCV_MLU_ARGS', '-DNDEBUG ')
mluops_includes = [] mluops_includes = []
mluops_includes.append('-I' + mluops_includes.append('-I' +
os.path.abspath('./mlu-ops/bangc-ops')) os.path.abspath('./mlu-ops/bangc-ops'))
......
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