From 055a056c21df9bfc3895f175f601e2df66261bd8 Mon Sep 17 00:00:00 2001 From: bdf <36697723+defei-coder@users.noreply.github.com> Date: Fri, 24 Mar 2023 19:59:11 +0800 Subject: [PATCH] [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 Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 5a7afe86f..95d551bfc 100644 --- a/setup.py +++ b/setup.py @@ -368,7 +368,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'))