python3Packages.torch: explicitly include libopenmp on Darwin x86_64

2025-08-03 19:13:09 -07:00
parent b70a3f45af
commit c387b7908b
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;
    [