Skip to content

Commit 6143384

Browse files
authored
[MLU] Fix setup for build (#362)
1 parent e5fefef commit 6143384

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

plugin/torch/setup.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,8 @@
7979
torch_mlu_path = torch_mlu.__file__.split("__init__")[0]
8080
torch_mlu_lib_dir = os.path.join(torch_mlu_path, "csrc/lib/")
8181
torch_mlu_include_dir = os.path.join(torch_mlu_path, "csrc/")
82-
include_dirs += [f"{neuware_home_path}/include", torch_mlu_include_dir]
82+
torch_mlu_include_dir2 = os.path.join(torch_mlu_path, "csrc", "include")
83+
include_dirs += [f"{neuware_home_path}/include", torch_mlu_include_dir, torch_mlu_include_dir2]
8384
library_dirs += [f"{neuware_home_path}/lib64", torch_mlu_lib_dir]
8485
libs += ["cnrt", "cncl", "torch_mlu"]
8586
elif adaptor_flag == "-DUSE_METAX_ADAPTOR":

0 commit comments

Comments
 (0)