ROCM bazelrc

Author: wsmoses

deps/ReactantExtra/.bazelrc

@@ -14,7 +14,6 @@ build --define=allow_oversize_protos=true
 
 build -c opt
 
-
 build:cuda --repo_env TF_NEED_CUDA=1
 build:cuda --repo_env TF_NCCL_USE_STUB=1
 build:cuda --repo_env=HERMETIC_CUDA_VERSION="12.3.2"
@@ -28,3 +27,10 @@ build:cuda --@xla//xla/python:jax_cuda_pip_rpaths=true
 # Default hermetic CUDA and CUDNN versions.
 build:cuda --@local_config_cuda//cuda:include_cuda_libs=true
 build:cuda --@local_config_cuda//:cuda_compiler=nvcc
+
+
+build:rocm --repo_env TF_NEED_ROCM=1
+build:rocm --define=using_rocm=true
+build:rocm --define=using_rocm_hipcc=true
+build:rocm --action_env TF_ROCM_AMDGPU_TARGETS="gfx900,gfx906,gfx908,gfx90a,gfx1030"
+