Skip to content

GitLab

Unverified Commit 42e7e2ee authored by Qiza-lyhm's avatar Qiza-lyhm Committed by GitHub
Browse files

[Fix] Add MLU kernels to be compiled (#1885) 

- Add MLU operator kernels.
parent 03b38a46
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
setup.py
View file @ 42e7e2ee
...@@ -301,7 +301,8 @@ def get_extensions(): ...@@ -301,7 +301,8 @@ def get_extensions():
extra_compile_args['cncc'] = [mlu_args] if mlu_args else [] extra_compile_args['cncc'] = [mlu_args] if mlu_args else []
op_files = glob.glob('./mmcv/ops/csrc/pytorch/*.cpp') + \ op_files = glob.glob('./mmcv/ops/csrc/pytorch/*.cpp') + \
glob.glob('./mmcv/ops/csrc/pytorch/cpu/*.cpp') + \ glob.glob('./mmcv/ops/csrc/pytorch/cpu/*.cpp') + \
glob.glob('./mmcv/ops/csrc/pytorch/mlu/*.cpp') glob.glob('./mmcv/ops/csrc/pytorch/mlu/*.cpp') + \
glob.glob('./mmcv/ops/csrc/common/mlu/*.mlu')
extension = MLUExtension extension = MLUExtension
include_dirs.append(os.path.abspath('./mmcv/ops/csrc/common')) include_dirs.append(os.path.abspath('./mmcv/ops/csrc/common'))
include_dirs.append(os.path.abspath('./mmcv/ops/csrc/common/mlu')) include_dirs.append(os.path.abspath('./mmcv/ops/csrc/common/mlu'))
......
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