Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
test_gpu.sh 258 Bytes
Newer Older
huchen's avatar
Init colossal, worldsize=4, 3d test需要8dcu  
huchen committed
1
2
3
4
5
6 
source /opt/dtk-22.04.2/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_zero_tensor_parallel/