Unverified Commit 56291f2e authored by Pavol Rusnak's avatar Pavol Rusnak Committed by GitHub
Browse files

torch: enable autoAddCudaCompatRunpath on Jetsons (#378960)

parents 2c704b65 94ad953c

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

+3 −0
Original line number Diff line number Diff line
@@ -150,6 +150,8 @@ let 
  supportedCudaCapabilities = lists.intersectLists cudaFlags.cudaCapabilities supportedTorchCudaCapabilities;

  unsupportedCudaCapabilities = lists.subtractLists supportedCudaCapabilities cudaFlags.cudaCapabilities;



  isCudaJetson = cudaSupport && cudaPackages.cudaFlags.isJetsonBuild;



  # Use trivial.warnIf to print a warning if any unsupported GPU targets are specified.

  gpuArchWarner =

    supported: unsupported:
@@ -456,6 +458,7 @@ buildPythonPackage rec { 
        cuda_nvcc

      ]

    )

    ++ lib.optionals isCudaJetson [ cudaPackages.autoAddCudaCompatRunpath ]

    ++ lib.optionals rocmSupport [ rocmtoolkit_joined ];



  buildInputs =