Commit df465820 authored by limm's avatar limm
Browse files

correction of bracket errors

parent 53bb5813
...@@ -276,7 +276,7 @@ def get_extensions(): ...@@ -276,7 +276,7 @@ def get_extensions():
elif torch.cuda.is_available() or os.getenv('FORCE_CUDA', '0') == '1': elif torch.cuda.is_available() or os.getenv('FORCE_CUDA', '0') == '1':
define_macros += [('MMCV_WITH_CUDA', None)] define_macros += [('MMCV_WITH_CUDA', None)]
if is_rocm_pytorch: if is_rocm_pytorch:
define_macros += [('HIP_DIFF', None) define_macros += [('HIP_DIFF', None)]
cuda_args = os.getenv('MMCV_CUDA_ARGS') cuda_args = os.getenv('MMCV_CUDA_ARGS')
extra_compile_args['nvcc'] = [cuda_args] if cuda_args else [] extra_compile_args['nvcc'] = [cuda_args] if cuda_args else []
op_files = glob.glob('./mmcv/ops/csrc/pytorch/*.cpp') + \ op_files = glob.glob('./mmcv/ops/csrc/pytorch/*.cpp') + \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment