diff --git a/tao/.bazelrc b/tao/.bazelrc index c4c91484a95..7bdcce6b37c 100644 --- a/tao/.bazelrc +++ b/tao/.bazelrc @@ -37,7 +37,7 @@ build:disc --config=release_base build:disc_cpu --config=disc --cxxopt=-DTAO_CPU_ONLY build:disc_x86 --config=disc_cpu --config=release_cpu_linux --cxxopt=-DTAO_X86 --define disc_x86=true -build:disc_aarch64 --config=disc_cpu --cxxopt=-DTAO_AARCH64 --define disc_aarch64=true --linkopt="-Xlinker --stub-group-size -Xlinker 10000000" +build:disc_aarch64 --config=disc_cpu --cxxopt=-DTAO_AARCH64 --define disc_aarch64=true --linkopt="-Xlinker --stub-group-size -Xlinker 10000000" --experimental_local_memory_estimate --local_ram_resources=HOST_RAM*.5 --local_cpu_resources=HOST_CPUS*.5 build:disc_cuda --config=disc --config=cuda build:disc_dcu --config=disc --config=dcu build:disc_rocm --config=disc --config=rocm diff --git a/tao_compiler/.bazelrc.user b/tao_compiler/.bazelrc.user index 2457c353e6e..8f5985e801f 100644 --- a/tao_compiler/.bazelrc.user +++ b/tao_compiler/.bazelrc.user @@ -3,7 +3,7 @@ build:ci_build --noshow_loading_progress --show_progress_rate_limit=600 --jobs=1 build:disc --define framework_shared_object=false --experimental_multi_threaded_digest build:disc_cpu --config=disc --cxxopt=-DTAO_CPU_ONLY build:disc_x86 --config=disc_cpu --config=release_cpu_linux --cxxopt=-DTAO_X86 --define disc_x86=true -build:disc_aarch64 --config=disc_cpu --config=mkl_aarch64 --cxxopt=-DTAO_AARCH64 --define disc_aarch64=true --linkopt="-Xlinker --stub-group-size -Xlinker 10000000" +build:disc_aarch64 --config=disc_cpu --config=mkl_aarch64 --cxxopt=-DTAO_AARCH64 --define disc_aarch64=true --linkopt="-Xlinker --stub-group-size -Xlinker 10000000" --experimental_local_memory_estimate --local_ram_resources=HOST_RAM*.5 --local_cpu_resources=HOST_CPUS*.5 build:disc_cuda --config=disc --config=cuda build:disc_dcu --config=disc --config=dcu diff --git a/tensorflow_blade/.bazelrc b/tensorflow_blade/.bazelrc index 4dfc84ad1a4..e0bfe8b5de3 100644 --- a/tensorflow_blade/.bazelrc +++ b/tensorflow_blade/.bazelrc @@ -69,6 +69,9 @@ build:disc_aarch64 --cxxopt=-DTAO_AARCH64 build:disc_aarch64 --define disc_aarch64=true build:disc_aarch64 --linkopt="-Xlinker --stub-group-size -Xlinker 10000000" build:disc_aarch64 --action_env BUILD_WITH_AARCH64=1 +build:disc_aarch64 --experimental_local_memory_estimate +build:disc_aarch64 --local_ram_resources=HOST_RAM*.5 +build:disc_aarch64 --local_cpu_resources=HOST_CPUS*.5 build:disc_cuda --config=disc --config=cuda build:disc_dcu --config=disc --config=dcu