From 66575be077b32fc10780fe604343e395eee05272 Mon Sep 17 00:00:00 2001 From: yao_yf Date: Wed, 9 Mar 2022 17:05:02 +0800 Subject: [PATCH] raises exception if no avaible strategies for matmul --- mindspore/ccsrc/frontend/parallel/ops_info/matmul_info.cc | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/mindspore/ccsrc/frontend/parallel/ops_info/matmul_info.cc b/mindspore/ccsrc/frontend/parallel/ops_info/matmul_info.cc index 7ec574bcaa7..c04b509523a 100644 --- a/mindspore/ccsrc/frontend/parallel/ops_info/matmul_info.cc +++ b/mindspore/ccsrc/frontend/parallel/ops_info/matmul_info.cc @@ -488,6 +488,9 @@ Status MatMulBase::GenerateStrategiesNotPower2(int64_t stage_id, size_t dev_num_ continue; } } + if (strategy_cost_.empty()) { + MS_LOG(EXCEPTION) << "No available strategy."; + } return SUCCESS; } @@ -523,6 +526,9 @@ Status MatMulBase::GenerateStrategies(int64_t stage_id) { continue; } } + if (strategy_cost_.empty()) { + MS_LOG(EXCEPTION) << "No available strategy."; + } return SUCCESS; } auto dev_num_not_2_power = dev_num / (dev_num - dev_num_2_power);