python3Packages.tinygrad: fix CUDA patch (#442994)
This commit is contained in:
@@ -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 \
|
||||
|
||||
Reference in New Issue
Block a user