python3Packages.torch: explicitly include libopenmp on Darwin x86_64 (#430808)

2025-08-04 18:55:56 -04:00
parent bd4291eea1 c387b7908b
commit a4a86ff877
1 changed files with 2 additions and 2 deletions
@@ -526,9 +526,9 @@ buildPythonPackage rec {
    blas
    blas.provider
  ]
-  # Including openmp leads to two copies being used. This segfaults on ARM.
+  # Including openmp leads to two copies being used on ARM, which segfaults.
  # https://github.com/pytorch/pytorch/issues/149201#issuecomment-2776842320
-  # ++ lib.optionals stdenv.cc.isClang [ llvmPackages.openmp ]
+  ++ lib.optionals (stdenv.cc.isClang && !stdenv.hostPlatform.isAarch64) [ llvmPackages.openmp ]
  ++ lib.optionals cudaSupport (
    with cudaPackages;
    [