Skip to content

Commit

Permalink
Turn off SHARP for DALI L2 tests
Browse files Browse the repository at this point in the history
- due some problems with CI runners turn of SHARP that is not
  needed for the tests

Signed-off-by: Janusz Lisiecki <[email protected]>
  • Loading branch information
JanuszL committed Aug 30, 2024
1 parent bdcf160 commit dee3bff
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions qa/TL2_FW_iterators_perf/test_pytorch.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit dee3bff

Please sign in to comment.