diff --git a/include/api/context.h b/include/api/context.h index c815b73e52e..93ed2706177 100644 --- a/include/api/context.h +++ b/include/api/context.h @@ -469,7 +469,6 @@ class MS_API AscendDeviceInfo : public DeviceInfoContext { using Ascend310DeviceInfo = AscendDeviceInfo; using Ascend910DeviceInfo = AscendDeviceInfo; -using Ascend710DeviceInfo = AscendDeviceInfo; void AscendDeviceInfo::SetInsertOpConfigPath(const std::string &cfg_path) { SetInsertOpConfigPath(StringToChar(cfg_path)); diff --git a/mindspore/lite/test/st/scripts/ascend/run_benchmark_ascend.sh b/mindspore/lite/test/st/scripts/ascend/run_benchmark_ascend.sh index c803027e857..d8362af1ca4 100644 --- a/mindspore/lite/test/st/scripts/ascend/run_benchmark_ascend.sh +++ b/mindspore/lite/test/st/scripts/ascend/run_benchmark_ascend.sh @@ -130,8 +130,8 @@ echo 'run Ascend logs: ' > ${run_ascend_log_file} echo "Start to run benchmark in ${backend}, device id ${device_id}..." if [[ ${backend} =~ "ascend310" ]]; then ascend_device=Ascend310 -elif [[ ${backend} =~ "ascend710" ]]; then - ascend_device=Ascend710 +elif [[ ${backend} =~ "ascend310P" ]]; then + ascend_device=Ascend310P else echo "${backend} is not support." exit 1 diff --git a/mindspore/lite/tools/benchmark/benchmark_base.cc b/mindspore/lite/tools/benchmark/benchmark_base.cc index 92066b224b1..7515abbc765 100644 --- a/mindspore/lite/tools/benchmark/benchmark_base.cc +++ b/mindspore/lite/tools/benchmark/benchmark_base.cc @@ -295,7 +295,7 @@ int BenchmarkBase::CheckThreadNumValid() { int BenchmarkBase::CheckDeviceTypeValid() { if (flags_->device_ != "CPU" && flags_->device_ != "GPU" && flags_->device_ != "NPU" && - flags_->device_ != "Ascend310" && flags_->device_ != "Ascend710") { + flags_->device_ != "Ascend310" && flags_->device_ != "Ascend310P") { MS_LOG(ERROR) << "Device type:" << flags_->device_ << " is not supported."; std::cerr << "Device type:" << flags_->device_ << " is not supported." << std::endl; return RET_ERROR; diff --git a/mindspore/lite/tools/benchmark/benchmark_base.h b/mindspore/lite/tools/benchmark/benchmark_base.h index f32f7d25cd9..8e4acfcb93e 100644 --- a/mindspore/lite/tools/benchmark/benchmark_base.h +++ b/mindspore/lite/tools/benchmark/benchmark_base.h @@ -121,7 +121,7 @@ class MS_API BenchmarkFlags : public virtual FlagParser { AddFlag(&BenchmarkFlags::model_type_, "modelType", "Input model type. MindIR | MindIR_Lite", "MindIR"); AddFlag(&BenchmarkFlags::in_data_file_, "inDataFile", "Input data file, if not set, use random input", ""); AddFlag(&BenchmarkFlags::config_file_, "configFile", "Config file", ""); - AddFlag(&BenchmarkFlags::device_, "device", "CPU | GPU | NPU | Ascend310 | Ascend710", "CPU"); + AddFlag(&BenchmarkFlags::device_, "device", "CPU | GPU | NPU | Ascend310 | Ascend310P", "CPU"); AddFlag(&BenchmarkFlags::cpu_bind_mode_, "cpuBindMode", "Input 0 for NO_BIND, 1 for HIGHER_CPU, 2 for MID_CPU.", 1); // MarkPerformance AddFlag(&BenchmarkFlags::loop_count_, "loopCount", "Run loop count", 10); diff --git a/mindspore/lite/tools/benchmark/benchmark_unified_api.cc b/mindspore/lite/tools/benchmark/benchmark_unified_api.cc index 4f9f1c67b0f..66de1aca208 100644 --- a/mindspore/lite/tools/benchmark/benchmark_unified_api.cc +++ b/mindspore/lite/tools/benchmark/benchmark_unified_api.cc @@ -426,7 +426,7 @@ int BenchmarkUnifiedApi::InitMSContext(const std::shared_ptr device_list.push_back(npu_device_info); } - if (flags_->device_ == "Ascend310" || flags_->device_ == "Ascend710") { + if (flags_->device_ == "Ascend310" || flags_->device_ == "Ascend310P") { uint32_t device_id = 0; auto device_id_env = std::getenv("ASCEND_DEVICE_ID"); if (device_id_env != nullptr) {