hccl fix (#3236)

open-mmlab · Feb 10, 2025 · a4a884d · a4a884d
1 parent 53e6b94
commit a4a884d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -408,6 +408,11 @@ def get_mluops_version(file_path):
                         'torch_npu').submodule_search_locations[0] +
                     '/include/third_party/acl/inc'
                 ]
+                extra_compile_args['cxx'] += [
+                    '-I' + importlib.util.find_spec(
+                        'torch_npu').submodule_search_locations[0] +
+                    '/include/third_party/hccl/inc'
+                ]
                 define_macros += [('MMCV_WITH_NPU', None)]
                 extension = NpuExtension
                 if parse_version(torch.__version__) < parse_version('2.1.0'):