Home
last modified time | relevance | path

Searched refs:DEVICE_NUM (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/tests/st/tbe_networks/
Drun_data_parallel.sh18 export DEVICE_NUM=2
19 export RANK_SIZE=$DEVICE_NUM
21 export DISTRIBUTION_FILE=$BASE_PATH/tdt${DEVICE_NUM}p/tdt_
22 export MINDSPORE_HCCL_CONFIG_PATH=$BASE_PATH/hccl_${DEVICE_NUM}p.json
24 for((i=0;i<$DEVICE_NUM;i++))
35 python resnet_cifar.py --run_distribute=1 --device_num=$DEVICE_NUM --epoch_size=10 >log 2>&1 &
/third_party/mindspore/tests/st/auto_parallel/
Drun_onehot_model_parallel.sh19 export DEVICE_NUM=8
20 export RANK_SIZE=$DEVICE_NUM
24 export MINDSPORE_HCCL_CONFIG_PATH=$CONFIG_PATH/hccl/rank_table_${DEVICE_NUM}p.json
27 for((i=0; i<$DEVICE_NUM; i++)); do
40 for((i=0; i<${DEVICE_NUM}; i++)); do
Drun_auto_parallel_loss_expand.sh19 export DEVICE_NUM=8
20 export RANK_SIZE=$DEVICE_NUM
23 export MINDSPORE_HCCL_CONFIG_PATH=$CONFIG_PATH/hccl/rank_table_${DEVICE_NUM}p.json
26 for((i=0; i<$DEVICE_NUM; i++)); do
39 for((i=0; i<${DEVICE_NUM}; i++)); do
Drun_auto_parallel_resnet50_expand_loss.sh19 export DEVICE_NUM=8
20 export RANK_SIZE=$DEVICE_NUM
23 export MINDSPORE_HCCL_CONFIG_PATH=$CONFIG_PATH/hccl/rank_table_${DEVICE_NUM}p.json
26 for((i=0; i<$DEVICE_NUM; i++)); do
39 for((i=0; i<${DEVICE_NUM}; i++)); do
Drun_optimizer_parallel.sh19 export DEVICE_NUM=4
20 export RANK_SIZE=$DEVICE_NUM
23 export MINDSPORE_HCCL_CONFIG_PATH=$CONFIG_PATH/hccl/rank_tabel_4p/rank_table_${DEVICE_NUM}p_1.json
28 for((i=0; i<$DEVICE_NUM; i++)); do
41 for((i=0; i<${DEVICE_NUM}; i++)); do
Drun_parallel_strategy_search.sh19 export DEVICE_NUM=8
20 export RANK_SIZE=$DEVICE_NUM
23 export MINDSPORE_HCCL_CONFIG_PATH=$CONFIG_PATH/hccl/rank_table_${DEVICE_NUM}p.json
28 for((i=0; i<$DEVICE_NUM; i++)); do
41 for((i=0; i<${DEVICE_NUM}; i++)); do
Drun_parallel_recursive_strategy_search.sh19 export DEVICE_NUM=8
20 export RANK_SIZE=$DEVICE_NUM
23 export MINDSPORE_HCCL_CONFIG_PATH=$CONFIG_PATH/hccl/rank_table_${DEVICE_NUM}p.json
28 for((i=0; i<$DEVICE_NUM; i++)); do
41 for((i=0; i<${DEVICE_NUM}; i++)); do
/third_party/mindspore/tests/st/broadcast/
Drun_broadcast_auto_parallel.sh22 export DEVICE_NUM=8
23 export RANK_SIZE=$DEVICE_NUM
26 export MINDSPORE_HCCL_CONFIG_PATH=$CONFIG_PATH/hccl/rank_table_${DEVICE_NUM}p.json
29 for ((i = 0; i < $DEVICE_NUM; i++)); do
42 for ((i = 0; i < ${DEVICE_NUM}; i++)); do
/third_party/mindspore/tests/st/hcom/
Drun_hcom_sparsetensor.sh22 export DEVICE_NUM=8
23 export RANK_SIZE=$DEVICE_NUM
26 export MINDSPORE_HCCL_CONFIG_PATH=$CONFIG_PATH/hccl/rank_table_${DEVICE_NUM}p.json
29 for ((i = 0; i < $DEVICE_NUM; i++)); do
42 for ((i = 0; i < ${DEVICE_NUM}; i++)); do
/third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/
Drun_wide_and_deep_auto_parallel.sh19 export DEVICE_NUM=8
20 export RANK_SIZE=$DEVICE_NUM
22 export MINDSPORE_HCCL_CONFIG_PATH=$CONFIG_PATH/hccl/rank_table_${DEVICE_NUM}p.json
43 for((i=0; i<$DEVICE_NUM; i++)); do
55 for((i=0; i<${DEVICE_NUM}; i++)); do
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dbatchnorm_info.cc229 Attr attr_device_num = std::make_pair(DEVICE_NUM, device_num); in InferReplaceOps()
Dops_utils.h227 constexpr char DEVICE_NUM[] = "device_num"; variable