chore: change npu id array

This commit is contained in:
wql 2024-09-18 16:18:37 +08:00
parent cba294f74d
commit 43f4c8f914
2 changed files with 4 additions and 3 deletions

View File

@ -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() {

View File

@ -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"