Skip to content

Commit dab7ff8

Browse files
committed
add openmp path.
1 parent 680d20d commit dab7ff8

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

torch/_inductor/cpp_builder.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -910,8 +910,15 @@ def _get_python_related_args() -> tuple[list[str], list[str]]:
910910
str(
911911
(
912912
Path(sysconfig.get_path("include", scheme="nt")).parent / "libs"
913-
).absolute()
914-
)
913+
).absolute() # python[ver].lib
914+
),
915+
str(
916+
(
917+
Path(sysconfig.get_path("include", scheme="nt")).parent
918+
/ "Library"
919+
/ "lib"
920+
).absolute() # install python librarys location, such as intel-openmp
921+
),
915922
]
916923
else:
917924
python_lib_path = [sysconfig.get_config_var("LIBDIR")]

0 commit comments

Comments
 (0)