From dee3bff73032908aa2dcc915ba1632024aab6d64 Mon Sep 17 00:00:00 2001 From: Janusz Lisiecki Date: Fri, 30 Aug 2024 18:26:29 +0200 Subject: [PATCH] Turn off SHARP for DALI L2 tests - due some problems with CI runners turn of SHARP that is not needed for the tests Signed-off-by: Janusz Lisiecki --- qa/TL2_FW_iterators_perf/test_pytorch.sh | 3 +++ 1 file changed, 3 insertions(+) diff --git a/qa/TL2_FW_iterators_perf/test_pytorch.sh b/qa/TL2_FW_iterators_perf/test_pytorch.sh index 5df1a4a74e..0d7df655f9 100755 --- a/qa/TL2_FW_iterators_perf/test_pytorch.sh +++ b/qa/TL2_FW_iterators_perf/test_pytorch.sh @@ -9,6 +9,9 @@ do_once() { } test_body() { + # turn off SHARP to avoid NCCL errors + export NCCL_NVLS_ENABLE=0 + for fw in "pytorch"; do python test_RN50_data_fw_iterators.py --framework ${fw} --gpus ${NUM_GPUS} -b 13 \ --workers 3 --prefetch 2 --epochs 3