chore: change npu id array
This commit is contained in:
parent
cba294f74d
commit
43f4c8f914
|
@ -7,9 +7,9 @@ print_to_screen=$3
|
|||
output_file_path="${output_dir}/gpu_status_$(date +"%Y%m%d%H%M%S").json"
|
||||
|
||||
device_cnt=$(npu-smi info | grep '310P3' | wc -l)
|
||||
chip_id_arr_from_info=(0 1 0 1 0 1 0 1)
|
||||
npu_id_arr_from_info=(2 2 3 3 5 5 6 6)
|
||||
npu_id_arr=(2 3 5 6)
|
||||
chip_id_arr_from_info=(0 0 0 0 0 0 0 0)
|
||||
npu_id_arr_from_info=(0 1 2 3 4 5 6 7)
|
||||
npu_id_arr=(0 1 2 3 4 5 6 7)
|
||||
|
||||
|
||||
get_power_dissipation() {
|
||||
|
|
|
@ -2,6 +2,7 @@
|
|||
|
||||
|
||||
# bash run_once.sh lora_sft Qwen-7B 8 50
|
||||
# bash run_once.sh inference Qwen-7B 8 50
|
||||
run_type="$1"
|
||||
model="$2"
|
||||
gpu_cnt="$3"
|
||||
|
|
Loading…
Reference in New Issue