diff options
| author | Sreenivasa Murthy Kolam <sreenivasamurthy.kolam@amd.com> | 2024-03-08 03:53:40 +0530 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-03-07 15:23:40 -0700 |
| commit | c6fcb1068f0f717a40d4ab97353ffa7fe5d885c5 (patch) | |
| tree | b56ad0b155a72627c3cb6c9c5b5e7cf72f1225a7 | |
| parent | 54ac3e72ed8bc1dbdf24d168cae8f9af23141859 (diff) | |
| download | spack-c6fcb1068f0f717a40d4ab97353ffa7fe5d885c5.tar.gz spack-c6fcb1068f0f717a40d4ab97353ffa7fe5d885c5.tar.bz2 spack-c6fcb1068f0f717a40d4ab97353ffa7fe5d885c5.tar.xz spack-c6fcb1068f0f717a40d4ab97353ffa7fe5d885c5.zip | |
Enable tensorflow-2.11 support for ROCm (#38520)
* enable tensorflow-2.11 support for ROCm
* add latest sha for mesa and limit the patches to older version.similar
changes in #37910 to enable gitlab-ci pass
* address review commemts
| -rw-r--r-- | var/spack/repos/builtin/packages/py-libclang/package.py | 4 | ||||
| -rw-r--r-- | var/spack/repos/builtin/packages/py-tensorflow/package.py | 9 |
2 files changed, 10 insertions, 3 deletions
diff --git a/var/spack/repos/builtin/packages/py-libclang/package.py b/var/spack/repos/builtin/packages/py-libclang/package.py index 0ce9997290..ce39c4b9b7 100644 --- a/var/spack/repos/builtin/packages/py-libclang/package.py +++ b/var/spack/repos/builtin/packages/py-libclang/package.py @@ -16,6 +16,8 @@ class PyLibclang(PythonPackage): license("Apache-2.0") + version("16.0.0", sha256="a3eae57519209ed6fca4e76425f3159e54a08cbb2918d92a7a35640d4c28ec07") + version("15.0.6.1", sha256="f8ac6e30868e9eb92bb1001920230381565f9a3cf415411d3b67bb2339640d81") version("14.0.6", sha256="3666679d9f23270a230a4d4dae49bc48fc2515c272ff5855b2618e23daa50100") version("14.0.1", sha256="58a255381d6360aca8d4978c8bb2e6be55ac0bdd18bc10372da0febe0a7f9472") version("13.0.0", sha256="2638e81fe3976f4ad487dc6094dacf306dcb161e11b0830391d58d1ae1e05c80") @@ -28,7 +30,7 @@ class PyLibclang(PythonPackage): depends_on("python@2.7:2.8,3.3:", type=("build", "run")) depends_on("py-setuptools", type="build") - for ver in ["9", "10", "11", "13", "14"]: + for ver in ["9", "10", "11", "13", "14", "15", "16"]: depends_on("llvm+clang@" + ver, when="@" + ver, type="build") def patch(self): diff --git a/var/spack/repos/builtin/packages/py-tensorflow/package.py b/var/spack/repos/builtin/packages/py-tensorflow/package.py index 5551a15b4a..20fdd358a6 100644 --- a/var/spack/repos/builtin/packages/py-tensorflow/package.py +++ b/var/spack/repos/builtin/packages/py-tensorflow/package.py @@ -26,6 +26,11 @@ class PyTensorflow(Package, CudaPackage, ROCmPackage, PythonExtension): version("2.12.1", sha256="6bc4600cc0b88e9e40f1800096f5bddbbd3b6e5527a030dea631b87f2ae46b5b") version("2.12.0", sha256="c030cb1905bff1d2446615992aad8d8d85cbe90c4fb625cee458c63bf466bc8e") version("2.11.1", sha256="624ed1cc170cdcc19e8a15d8cdde989a9a1c6b0534c90b38a6b2f06fb2963e5f") + version( + "2.11.0-rocm-enhanced", + sha256="0c4ee8d83bc72215cbc1a5cd3e88cde1a9cf7304237d3e3d8d105ff09827d903", + url="https://github.com/ROCmSoftwarePlatform/tensorflow-upstream/archive/refs/tags/v2.11.0-rocm-enhanced.tar.gz", + ) version("2.11.0", sha256="99c732b92b1b37fc243a559e02f9aef5671771e272758aa4aec7f34dc92dac48") version("2.10.1", sha256="622a92e22e6f3f4300ea43b3025a0b6122f1cc0e2d9233235e4c628c331a94a3") version("2.10.0", sha256="b5a1bb04c84b6fe1538377e5a1f649bb5d5f0b2e3625a3c526ff3a8af88633e8") @@ -323,8 +328,8 @@ class PyTensorflow(Package, CudaPackage, ROCmPackage, PythonExtension): conflicts("platform=darwin target=aarch64:", when="@:2.4") # https://github.com/tensorflow/tensorflow/pull/39225 conflicts("target=aarch64:", when="@:2.2") - conflicts("~rocm", when="@2.7.4-rocm-enhanced") - conflicts("+rocm", when="@:2.7.4-a,2.7.4.0:") + conflicts("~rocm", when="@2.7.4-rocm-enhanced,2.11.0-rocm-enhanced") + conflicts("+rocm", when="@:2.7.4-a,2.7.4.0:2.11.0-a,2.11.0.0:") # wheel 0.40 upgrades vendored packaging, trips over tensorflow-io-gcs-filesystem identifier conflicts("^py-wheel@0.40:", when="@2.11:2.13") |