python3Packages.torch: propagate pybind11

2023-11-08 16:35:24 -06:00
parent 956abdcaee
commit 6d7ae4f426
1 changed files with 4 additions and 1 deletions
@@ -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 ];