Unverified Commit 94ad953c authored by Conroy Cheers's avatar Conroy Cheers Committed by Pavol Rusnak
Browse files

torch: enable autoAddCudaCompatRunpath on Jetsons

parent 508be1c4

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 =