Skip to content

Commit 7a0c9a5

Browse files
author
Anurag Dixit
committed
feat: Add TF32 override flag in bazelrc for CI-Testing
Signed-off-by: Anurag Dixit <[email protected]>
1 parent 119993b commit 7a0c9a5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.bazelrc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,8 +35,8 @@ build:pre_cxx11_abi --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0"
3535
build:pre_cxx11_abi --linkopt="-D_GLIBCXX_USE_CXX11_ABI=0"
3636
build:pre_cxx11_abi --define=abi=pre_cxx11_abi
3737

38-
build:ci_testing --define=torchtrt_src=pre_built --cxxopt="-DDISABLE_TEST_IN_CI"
38+
build:ci_testing --define=torchtrt_src=pre_built --cxxopt="-DDISABLE_TEST_IN_CI" --action_env "NVIDIA_TF32_OVERRIDE=0"
3939
build:use_precompiled_torchtrt --define=torchtrt_src=pre_built
4040

41-
test:ci_testing --define=torchtrt_src=pre_built --cxxopt="-DDISABLE_TEST_IN_CI"
41+
test:ci_testing --define=torchtrt_src=pre_built --cxxopt="-DDISABLE_TEST_IN_CI" --action_env "NVIDIA_TF32_OVERRIDE=0"
4242
test:use_precompiled_torchtrt --define=torchtrt_src=pre_built

0 commit comments

Comments
 (0)