mirror of
https://github.com/NixOS/nixpkgs.git
synced 2026-10-10 22:30:59 +00:00
f69fec58da
Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01XsW3g2o9DFfbYnbdGEcaKP
13 lines
381 B
Diff
13 lines
381 B
Diff
--- a/op_builder/cpu/builder.py
|
|
+++ b/op_builder/cpu/builder.py
|
|
@@ -16,6 +16,9 @@
|
|
|
|
class CPUOpBuilder(OpBuilder):
|
|
|
|
+ def extra_ldflags(self):
|
|
+ return ['-lomp']
|
|
+
|
|
def builder(self):
|
|
from torch.utils.cpp_extension import CppExtension as ExtensionBuilder
|
|
include_dirs = [os.path.abspath(x) for x in self.strip_empty_entries(self.include_paths())]
|