OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 12:20:11 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

PR #32678: [ROCm] Use working sha256 for latest ROCm 7.0 docker image and fix test scripts

Browse Source 
Imported from GitHub PR https://github.com/openxla/xla/pull/32678

📝 Summary of Changes
- Fix sha256 of docker image to ensure CI is not broken due to malformed image
- Fix test scripts by passing ROCM_PATH to bazel sandbox via repo_env

🎯 Justification
Continued CI runs

🚀 Kind of Contribution
 🧪 Tests

Copybara import of the project:

--
3ca8114613d8e002c137f28bb6608639d08a724a by Harsha Havanur Shamsundara <harsha.havanurshamsundara@amd.com>:

[ROCm] Use working sha256 for latest ROCm 7.0 docker image

--
09ddfbdf205a6406cdd67e20671f41455fffe0f9 by Harsha HS <Harsha.HavanurShamsundara@amd.com>:

[ROCm] Add ROCM_PATH repo_env to test scripts

Merging this change closes #32678

PiperOrigin-RevId: 820582560
This commit is contained in:
Harsha H S 2025-10-17 01:16:43 -07:00 committed by TensorFlower Gardener
parent f573329cc6
commit 086937e138
3 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
third_party/xla/build_tools/rocm/platform/linux_x64/BUILD vendored
View File

@ -26,7 +26,8 @@ platform(
"@bazel_tools//tools/cpp:clang", "@bazel_tools//tools/cpp:clang",
], ],
exec_properties = { exec_properties = {
"container-image": "docker://rocm/tensorflow-build:latest-jammy-python3.10-rocm6.4.0@sha256:9c2fbc861570735fc86dbf39dd3cefef6a13fcf6ae08358271af52f1aebd4248", # SHA256 for rocm/tensorflow-build:latest-jammy-python3.10-rocm7.0.0
"container-image": "docker://rocm/tensorflow-build@sha256:7cd444ac48657fee2f5087fbda7766266704d3f8fb2299f681952ae4eabed060",
"OSFamily": "Linux", "OSFamily": "Linux",
}, },
) )

1
third_party/xla/build_tools/rocm/run_xla.sh vendored
View File

@ -72,5 +72,6 @@ bazel \
--action_env=TF_ROCM_AMDGPU_TARGETS=${AMD_GPU_GFX_ID} \ --action_env=TF_ROCM_AMDGPU_TARGETS=${AMD_GPU_GFX_ID} \
--action_env=XLA_FLAGS=--xla_gpu_force_compilation_parallelism=16 \ --action_env=XLA_FLAGS=--xla_gpu_force_compilation_parallelism=16 \
--action_env=XLA_FLAGS=--xla_gpu_enable_llvm_module_compilation_parallelism=true \ --action_env=XLA_FLAGS=--xla_gpu_enable_llvm_module_compilation_parallelism=true \
--repo_env="ROCM_PATH=$ROCM_PATH" \
--run_under=//build_tools/ci:parallel_gpu_execute \ --run_under=//build_tools/ci:parallel_gpu_execute \
-- //xla/... -- //xla/...

1
third_party/xla/build_tools/rocm/run_xla_multi_gpu.sh vendored
View File

@ -90,6 +90,7 @@ bazel \
--action_env=XLA_FLAGS=--xla_gpu_force_compilation_parallelism=16 \ --action_env=XLA_FLAGS=--xla_gpu_force_compilation_parallelism=16 \
--action_env=XLA_FLAGS=--xla_gpu_enable_llvm_module_compilation_parallelism=true \ --action_env=XLA_FLAGS=--xla_gpu_enable_llvm_module_compilation_parallelism=true \
--action_env=NCCL_MAX_NCHANNELS=1 \ --action_env=NCCL_MAX_NCHANNELS=1 \
--repo_env="ROCM_PATH=$ROCM_PATH" \
-- //xla/tests:collective_ops_e2e_test \ -- //xla/tests:collective_ops_e2e_test \
//xla/tests:collective_ops_test \ //xla/tests:collective_ops_test \
//xla/tests:collective_pipeline_parallelism_test \ //xla/tests:collective_pipeline_parallelism_test \