Unverified Commit 6d7ae4f4 authored by Madoura's avatar Madoura
Browse files

python3Packages.torch: propagate pybind11

parent 956abdca

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

+4 −1
Original line number Diff line number Diff line
@@ -291,7 +291,7 @@ in buildPythonPackage rec { 
  ])

  ++ lib.optionals rocmSupport [ rocmtoolkit_joined ];



  buildInputs = [ blas blas.provider pybind11 ]

  buildInputs = [ blas blas.provider ]

    ++ lib.optionals stdenv.isLinux [ linuxHeaders_5_19 ] # TMP: avoid "flexible array member" errors for now

    ++ lib.optionals cudaSupport (with cudaPackages; [

      cuda_cccl.dev # <thrust/*>
@@ -345,6 +345,9 @@ in buildPythonPackage rec { 


    # ROCm build and `torch.compile` requires openai-triton

    openai-triton



    # torch/csrc requires `pybind11` at runtime

    pybind11

  ]

  ++ lib.optionals MPISupport [ mpi ]

  ++ lib.optionals rocmSupport [ rocmtoolkit_joined ];