commit a4a86ff87708953910f47db8d11ae5fc3137e222 · pyrox.dev/nixpkgs

pyrox.dev / nixpkgs

lol

fork atom

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

authored by Nick Cao and committed by GitHub 6 months ago a4a86ff8 bd4291ee

+2 -2

1 changed file

expand all

unified split

pkgs

development

python-modules

torch

source

default.nix

+2 -2

pkgs/development/python-modules/torch/source/default.nix

··· 526 blas 527 blas.provider 528 ] 529 - # Including openmp leads to two copies being used. This segfaults on ARM. 530 # https://github.com/pytorch/pytorch/issues/149201#issuecomment-2776842320 531 - # ++ lib.optionals stdenv.cc.isClang [ llvmPackages.openmp ] 532 ++ lib.optionals cudaSupport ( 533 with cudaPackages; 534 [

··· 526 blas 527 blas.provider 528 ] 529 + # Including openmp leads to two copies being used on ARM, which segfaults. 530 # https://github.com/pytorch/pytorch/issues/149201#issuecomment-2776842320 531 + ++ lib.optionals (stdenv.cc.isClang && !stdenv.hostPlatform.isAarch64) [ llvmPackages.openmp ] 532 ++ lib.optionals cudaSupport ( 533 with cudaPackages; 534 [