Commit f28f9bb4 authored by Nathan Ringo's avatar Nathan Ringo
Browse files

python3Packages.tinygrad: fix CUDA patch

parent 37f6a72a

pkgs/development/python-modules/tinygrad/default.nix

+2 −2
Original line number Diff line number Diff line
@@ -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 \