Skip to content

Commit 4d6e280

Browse files
committed
fix: update linker flag for OpenMP support in setup.py
1 parent d372cee commit 4d6e280

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
# check if openmp is available
1818
if torch.backends.openmp.is_available():
1919
extra_compile_args["cxx"] = ["-fopenmp"]
20-
extra_link_args.append("-lgomp")
20+
extra_link_args.append("-fopenmp")
2121

2222
setuptools.setup(
2323
name=NAME,

0 commit comments

Comments
 (0)