Unverified Commit 4288bd65 authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

python3Packages.tinygrad: fix CUDA patch (#442994)

parents ccc1abec f28f9bb4

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 \