test_gpu.sh

source /opt/dtk/env.sh
export LD_LIBRARY_PATH=/usr/local/lib/python3.7/site-packages/torch/lib/:$LD_LIBRARY_PATH

#export HIP_VISIBLE_DEVICES=0,1,2,3 
DATA=./cifar_dataset pytest  tests

#DATA=./cifar_dataset pytest -v tests/test_engine/test_engine/test_engine_naive_amp.py
#DATA=./cifar_dataset pytest -v tests/test_zero_data_parallel/test_zero_level_3.py
#DATA=./cifar_dataset pytest -v tests/test_zero_data_parallel/test_zero_level_2.py
#DATA=./cifar_dataset pytest -v tests/test_zero_data_parallel/test_zero_level_3.py
#DATA=./cifar_dataset pytest -v tests/test_zero_tensor_parallel/test_vit_2d_level_2.py