rocm

Author: wsmoses

deps/ReactantExtra/.bazelrc

@@ -44,8 +44,16 @@ build:cuda --@local_config_cuda//:cuda_compiler=nvcc
 # build:cuda --@local_config_nvshmem//:override_include_nvshmem_libs=true
 # build:cuda --@local_config_nvshmem//cuda:include_nvshmem_libs=true
 
-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"
 
+build:rocm --crosstool_top=@local_config_rocm//crosstool:toolchain
+build:rocm --define=using_rocm=true --define=using_rocm_hipcc=true
+build:rocm --repo_env TF_NEED_ROCM=1
+build:rocm --action_env TF_ROCM_AMDGPU_TARGETS="gfx900,gfx906,gfx908,gfx90a,gfx940,gfx941,gfx942,gfx1030,gfx1100,gfx1200,gfx1201"
+
+# Build with hipcc for ROCm and clang for the host.
+build:rocm --config=rocm_base
+build:rocm --action_env=TF_ROCM_CLANG="1"
+build:rocm --action_env=CLANG_COMPILER_PATH="/usr/lib/llvm-18/bin/clang"
+build:rocm --copt=-Wno-gnu-offsetof-extensions
+build:rocm --copt=-Qunused-arguments
+build:rocm --action_env=TF_HIPCC_CLANG="1"