Skip to content

Commit

Permalink
[Enhance] Add a default value for MMCV_MLU_ARGS (open-mmlab#2688)
Browse files Browse the repository at this point in the history
* add mlu_args

* add mlu_args

* Modify the code

---------

Co-authored-by: budefei <budefei@cambricon.com>
  • Loading branch information
2 people authored and CokeDong committed Apr 6, 2023
1 parent a3dba8e commit 9e1dc99
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -376,7 +376,7 @@ def get_mluops_version(file_path):
exit()

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.append('-I' +
os.path.abspath('./mlu-ops/bangc-ops'))
Expand Down

0 comments on commit 9e1dc99

Please sign in to comment.