From 66c5515174110ffab353fa86d9e07a4352c51eed Mon Sep 17 00:00:00 2001 From: bdf <36697723+defei-coder@users.noreply.github.com> Date: Fri, 24 Mar 2023 19:59:43 +0800 Subject: [PATCH] [Enhance] Add a default value for MMCV_MLU_ARGS (#2688) * add mlu_args * add mlu_args * Modify the code --------- Co-authored-by: budefei --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index c12a85358..8e8efdde7 100644 --- a/setup.py +++ b/setup.py @@ -369,7 +369,7 @@ def get_extensions(): 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'))