mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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

* add mlu_args

* add mlu_args

* Modify the code

---------

Co-authored-by: budefei <budefei@cambricon.com>
pull/2700/head
bdf 2023-03-24 19:59:43 +08:00 committed by GitHub
parent 4c4ba6cb17
commit 66c5515174
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@ -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'))