From d1d640e33141e8a1693c338a0f73a466b6c124ee Mon Sep 17 00:00:00 2001
From: deftdawg <deftdawg@gmail.com>
Date: Mon, 23 Dec 2024 11:49:38 -0500
Subject: [PATCH] Add compiler_hip fixes

@wozeparrot's fixes for compiler_hip.py
---
 pkgs/development/python-modules/tinygrad/default.nix | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/pkgs/development/python-modules/tinygrad/default.nix b/pkgs/development/python-modules/tinygrad/default.nix
index 7ae30a43b8aab..c9bc6325b48d2 100644
--- 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}/"
     '';