From c387b7908bd35d57b3e1df6e587dd4dd547c88bd Mon Sep 17 00:00:00 2001 From: Sarah Clark Date: Sun, 3 Aug 2025 15:55:05 -0700 Subject: [PATCH] python3Packages.torch: explicitly include libopenmp on Darwin x86_64 --- pkgs/development/python-modules/torch/source/default.nix | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pkgs/development/python-modules/torch/source/default.nix b/pkgs/development/python-modules/torch/source/default.nix index 5e16b0284c763..18ede02d80d98 100644 --- a/pkgs/development/python-modules/torch/source/default.nix +++ b/pkgs/development/python-modules/torch/source/default.nix @@ -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; [