lol

onnxruntime: add autoAddCudaCompatRunpath on Jetson (#357738)

authored by

Pavol Rusnak and committed by
GitHub
f9ffac5a 218681f7

+4 -2
+4 -2
pkgs/development/libraries/onnxruntime/default.nix
··· 81 81 hash = "sha256-9oORW0YlQ6SphqfbjcYb0dTlHc+1gzy9quH/Lj6By8Q="; 82 82 }; 83 83 84 - cutlass = fetchFromGitHub { 84 + cutlass = fetchFromGitHub { 85 85 owner = "NVIDIA"; 86 86 repo = "cutlass"; 87 87 tag = "v3.5.1"; ··· 95 95 hash = "sha256-YJdZ0cMtUncH5Z6TtAWBH0xtAIu2UcbjnVcCM4tfg20="; 96 96 }; 97 97 98 + isCudaJetson = cudaSupport && cudaPackages.cudaFlags.isJetsonBuild; 98 99 in 99 - 100 100 effectiveStdenv.mkDerivation rec { 101 101 pname = "onnxruntime"; 102 102 inherit src version; ··· 128 128 ]) ++ lib.optionals cudaSupport [ 129 129 cudaPackages.cuda_nvcc 130 130 cudaPackages.cudnn-frontend 131 + ] ++ lib.optionals isCudaJetson [ 132 + cudaPackages.autoAddCudaCompatRunpath 131 133 ]; 132 134 133 135 buildInputs = [