Add compiler_hip fixes

@wozeparrot's fixes for compiler_hip.py
khaneliman · Dec 23, 2024 · d1d640e · d1d640e
1 parent 2877041
commit d1d640e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pkgs/development/python-modules/tinygrad/default.nix b/pkgs/development/python-modules/tinygrad/default.nix
@@ -88,6 +88,12 @@ buildPythonPackage rec {
       substituteInPlace tinygrad/runtime/autogen/hip.py \
         --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/" "${rocmPackages.rocm-comgr}/"
     '';