diff --git a/.bazelrc b/.bazelrc index 56ade15da..103f0b345 100644 --- a/.bazelrc +++ b/.bazelrc @@ -294,6 +294,8 @@ build:tpu --define=enable_mlir_bridge=true 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 --define=xnn_enable_avxvnniint8=false +build:rocm --define=xnn_enable_avx512fp16=false build:rocm --repo_env TF_NEED_ROCM=1 build:rocm_clang_official --config=rocm