We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e5fefef commit 6143384Copy full SHA for 6143384
1 file changed
plugin/torch/setup.py
@@ -79,7 +79,8 @@
79
torch_mlu_path = torch_mlu.__file__.split("__init__")[0]
80
torch_mlu_lib_dir = os.path.join(torch_mlu_path, "csrc/lib/")
81
torch_mlu_include_dir = os.path.join(torch_mlu_path, "csrc/")
82
- include_dirs += [f"{neuware_home_path}/include", torch_mlu_include_dir]
+ 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]
84
library_dirs += [f"{neuware_home_path}/lib64", torch_mlu_lib_dir]
85
libs += ["cnrt", "cncl", "torch_mlu"]
86
elif adaptor_flag == "-DUSE_METAX_ADAPTOR":
0 commit comments