1.x branch fix hccl compile

hust17yixuan · Feb 5, 2025 · ce4e654 · ce4e654
1 parent 5e2b9a7
commit ce4e654
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -446,6 +446,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'):