diff --git a/pkgs/development/python-modules/tinygrad/default.nix b/pkgs/development/python-modules/tinygrad/default.nix index e59579ea8b33..d0e0b2418b1e 100644 --- a/pkgs/development/python-modules/tinygrad/default.nix +++ b/pkgs/development/python-modules/tinygrad/default.nix @@ -107,8 +107,8 @@ buildPythonPackage rec { + lib.optionalString cudaSupport '' substituteInPlace tinygrad/runtime/support/compiler_cuda.py \ --replace-fail \ - ', "-I/usr/local/cuda/include", "-I/usr/include", "-I/opt/cuda/include/"' \ - ', "-I${lib.getDev cudaPackages.cuda_cudart}/include/"' + '"-I/usr/local/cuda/include", "-I/usr/include", "-I/opt/cuda/include"' \ + '"-I${lib.getDev cudaPackages.cuda_cudart}/include/"' '' + lib.optionalString rocmSupport '' substituteInPlace tinygrad/runtime/autogen/hip.py \