gador/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update RoCM substitutions for tinygrad (#367501)

Browse Source
This commit is contained in:
Gaétan Lepage
2024-12-23 22:31:47 +01:00
committed by GitHub
parent de1864217b d1d640e331
commit ef27bba8d5
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkgs/development/python-modules/tinygrad/default.nix
+8 -3
View File
@@ -86,11 +86,16 @@ buildPythonPackage rec {
''
+ lib.optionalString rocmSupport ''
substituteInPlace tinygrad/runtime/autogen/hip.py \
--replace-fail "/opt/rocm/lib/libamdhip64.so" "${rocmPackages.clr}/lib/libamdhip64.so" \
--replace-fail "/opt/rocm/lib/libhiprtc.so" "${rocmPackages.clr}/lib/libhiprtc.so" \
--replace-fail "/opt/rocm/" "${rocmPackages.clr}/"
substituteInPlace tinygrad/runtime/support/compiler_hip.py \
--replace-fail "/opt/rocm/include" "${rocmPackages.clr}/include"
substituteInPlace tinygrad/runtime/support/compiler_hip.py \
--replace-fail "/opt/rocm/llvm" "${rocmPackages.llvm.llvm}"
substituteInPlace tinygrad/runtime/autogen/comgr.py \
--replace-fail "/opt/rocm/lib/libamd_comgr.so" "${rocmPackages.rocm-comgr}/lib/libamd_comgr.so"
--replace-fail "/opt/rocm/" "${rocmPackages.rocm-comgr}/"
'';
build-system = [ setuptools ];