From a6a6ec2e89bfbb98110ec084b34bace7cf48408e Mon Sep 17 00:00:00 2001 From: budefei Date: Wed, 22 Mar 2023 12:54:45 +0800 Subject: [PATCH 1/3] add mlu_args --- setup.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/setup.py b/setup.py index c12a85358a..d1189bc691 100644 --- a/setup.py +++ b/setup.py @@ -369,7 +369,8 @@ def get_mluops_version(file_path): exit() define_macros += [('MMCV_WITH_MLU', None)] - mlu_args = os.getenv('MMCV_MLU_ARGS') + mlu_args = '-DNDEBUG ' + mlu_args = os.getenv('MMCV_MLU_ARGS') if os.getenv('MMCV_MLU_ARGS') else mlu_args mluops_includes = [] mluops_includes.append('-I' + os.path.abspath('./mlu-ops/bangc-ops')) From ddb491d1ac849fa9708fcfa334caf28a0f5266ef Mon Sep 17 00:00:00 2001 From: budefei Date: Wed, 22 Mar 2023 15:01:08 +0800 Subject: [PATCH 2/3] add mlu_args --- setup.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/setup.py b/setup.py index c12a85358a..230fe3a679 100644 --- a/setup.py +++ b/setup.py @@ -369,7 +369,9 @@ def get_mluops_version(file_path): exit() define_macros += [('MMCV_WITH_MLU', None)] - mlu_args = os.getenv('MMCV_MLU_ARGS') + mlu_args = '-DNDEBUG ' + mlu_args = os.getenv('MMCV_MLU_ARGS') if os.getenv( + 'MMCV_MLU_ARGS') else mlu_args mluops_includes = [] mluops_includes.append('-I' + os.path.abspath('./mlu-ops/bangc-ops')) From cf6108ddaae0cf65080411aad2c3003fac0d8185 Mon Sep 17 00:00:00 2001 From: defei-coder <1755086696@qq.com> Date: Fri, 24 Mar 2023 19:50:52 +0800 Subject: [PATCH 3/3] Modify the code --- setup.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/setup.py b/setup.py index d1189bc691..8e8efdde7d 100644 --- a/setup.py +++ b/setup.py @@ -369,8 +369,7 @@ def get_mluops_version(file_path): exit() define_macros += [('MMCV_WITH_MLU', None)] - mlu_args = '-DNDEBUG ' - mlu_args = os.getenv('MMCV_MLU_ARGS') if os.getenv('MMCV_MLU_ARGS') else mlu_args + mlu_args = os.getenv('MMCV_MLU_ARGS', '-DNDEBUG ') mluops_includes = [] mluops_includes.append('-I' + os.path.abspath('./mlu-ops/bangc-ops'))