diff options
| author | afzpatel <122491982+afzpatel@users.noreply.github.com> | 2024-07-24 10:16:15 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-07-24 16:16:15 +0200 |
| commit | e529a454eb2a84388b3bab6154cf47ac1ecb6ef6 (patch) | |
| tree | 4bd9575d08268ca216e342c645fabc0e2f7f22a0 /var | |
| parent | 1b5dc396e3591011098814a460171f3834e95757 (diff) | |
| download | spack-e529a454eb2a84388b3bab6154cf47ac1ecb6ef6.tar.gz spack-e529a454eb2a84388b3bab6154cf47ac1ecb6ef6.tar.bz2 spack-e529a454eb2a84388b3bab6154cf47ac1ecb6ef6.tar.xz spack-e529a454eb2a84388b3bab6154cf47ac1ecb6ef6.zip | |
CI: add ML ROCm stack (#45302)
* add ML ROCm stack
* add suggested changes
* remove py-torch and py-tensorflow-estimator
* add TF_ROCM_AMDGPU_TARGETS env variable and remove packages from pipeline
* remove py-jax and py-xgboost
Diffstat (limited to 'var')
| -rw-r--r-- | var/spack/repos/builtin/packages/py-tensorflow/package.py | 1 |
1 files changed, 1 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 fd152cb89b..1d86e81fd0 100644 --- a/var/spack/repos/builtin/packages/py-tensorflow/package.py +++ b/var/spack/repos/builtin/packages/py-tensorflow/package.py @@ -562,6 +562,7 @@ class PyTensorflow(Package, CudaPackage, ROCmPackage, PythonExtension): for pkg_dep in rocm_dependencies: pkg_dep_cap = pkg_dep.upper().replace("-", "_") env.set(f"{pkg_dep_cap}_PATH", spec[pkg_dep].prefix) + env.set("TF_ROCM_AMDGPU_TARGETS", ",".join(self.spec.variants["amdgpu_target"].value)) else: env.set("TF_NEED_ROCM", "0") |