Fix tensilite path

ROCm · Nov 18, 2024 · a8ce7dc · a8ce7dc
1 parent 78d18f6
commit a8ce7dc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.bazelrc b/.bazelrc
@@ -283,6 +283,7 @@ build:rocm --crosstool_top=@local_config_rocm//crosstool:toolchain
 build:rocm --define=using_rocm_hipcc=true
 build:rocm --define=tensorflow_mkldnn_contraction_kernel=0
 build:rocm --repo_env TF_NEED_ROCM=1
+test:rocm --test_env=ROCBLAS_TENSILE_LIBPATH=/opt/rocm/lib/rocblas/library
 
 build:sycl --crosstool_top=@local_config_sycl//crosstool:toolchain
 build:sycl --define=using_sycl=true