Skip to content

Commit

Permalink
[Enhance] Add a default value for MMCV_MLU_ARGS (open-mmlab#2686)
Browse files Browse the repository at this point in the history
* 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>
  • Loading branch information
3 people committed Mar 24, 2023
1 parent 240767b commit 055a056
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 @@ -368,7 +368,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 055a056

Please sign in to comment.