zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

py-tensorflow: patch for cuBLAS error (#34279)

Browse Source
This commit is contained in:
Glenn Johnson 2022-12-03 10:17:12 -06:00 committed by GitHub
parent 8f5209063d
commit fe995542ab
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
var/spack/repos/builtin/packages/py-tensorflow/package.py
View File

@ -420,6 +420,19 @@ class PyTensorflow(Package, CudaPackage, ROCmPackage):
sha256="f9e26c544da729cfd376dbd3b096030e3777d3592459add1f3c78b1b9828d493", sha256="f9e26c544da729cfd376dbd3b096030e3777d3592459add1f3c78b1b9828d493",
when="@2.9:2.10.0", 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..." # Avoid build error: "no such package '@io_bazel_rules_docker..."
patch("io_bazel_rules_docker2.patch", when="@1.15:2.0") patch("io_bazel_rules_docker2.patch", when="@1.15:2.0")
# Avoide build error: "name 'new_http_archive' is not defined" # Avoide build error: "name 'new_http_archive' is not defined"