diff options
| author | Glenn Johnson <glenn-johnson@uiowa.edu> | 2022-12-03 10:17:12 -0600 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-12-03 09:17:12 -0700 |
| commit | fe995542ab615ad16e13e1143ea3d5a5b5a0eb94 (patch) | |
| tree | 75a1db87f303af2e346767fc3a1e398b94b4f02a | |
| parent | 8f5209063dbe97d08db61ebbbf06bf5d176275eb (diff) | |
| download | spack-fe995542ab615ad16e13e1143ea3d5a5b5a0eb94.tar.gz spack-fe995542ab615ad16e13e1143ea3d5a5b5a0eb94.tar.bz2 spack-fe995542ab615ad16e13e1143ea3d5a5b5a0eb94.tar.xz spack-fe995542ab615ad16e13e1143ea3d5a5b5a0eb94.zip | |
py-tensorflow: patch for cuBLAS error (#34279)
| -rw-r--r-- | var/spack/repos/builtin/packages/py-tensorflow/package.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-tensorflow/package.py b/var/spack/repos/builtin/packages/py-tensorflow/package.py index 1598f8c67b..b30899797a 100644 --- a/var/spack/repos/builtin/packages/py-tensorflow/package.py +++ b/var/spack/repos/builtin/packages/py-tensorflow/package.py @@ -420,6 +420,19 @@ class PyTensorflow(Package, CudaPackage, ROCmPackage): sha256="f9e26c544da729cfd376dbd3b096030e3777d3592459add1f3c78b1b9828d493", when="@2.9:2.10.0", ) + + # Version 2.10 produces an error related to cuBLAS: + # E tensorflow/stream_executor/cuda/cuda_blas.cc:2981] Unable to register + # cuBLAS factory: Attempting to register factory for plugin cuBLAS when one + # has already been registered + # See https://github.com/tensorflow/tensorflow/issues/57663 + # This is fixed for 2.11 but 2.10 needs the following patch. + patch( + "https://github.com/tensorflow/tensorflow/pull/56691.patch?full_index=1", + sha256="d635ea6d6c1571505871d0caba3e2cd939ea0f4aff972095d552913a8109def3", + when="@2.10", + ) + # Avoid build error: "no such package '@io_bazel_rules_docker..." patch("io_bazel_rules_docker2.patch", when="@1.15:2.0") # Avoide build error: "name 'new_http_archive' is not defined" |