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 (#2686) 

* add mlu_args

* Update setup.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

---------

Co-authored-by: budefei <budefei@cambricon.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
pull/3157/head
bdf 2023-03-24 19:59:11 +08:00 committed by GitHub
parent 240767bbb6
commit 055a056c21
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

@ -368,7 +368,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'))