Searched refs:stras (Results 1 – 6 of 6) sorted by relevance
51 Strategys stras = strategy->GetInputDim(); in CheckStrategy() local52 if (stras.size() != MATMUL_DDS_INPUTS_SIZE) { in CheckStrategy()56 for (auto stra : stras) { in CheckStrategy()65 if (stras[0][0] != stras[1][0] || num_heads_ % stras[0][0] != 0) { in CheckStrategy()66 MS_LOG(ERROR) << name_ << ": Invalid strategy. The strategys[0][0]:" << stras[0][0] in CheckStrategy()67 << " should be equal to strategys[1][0]:" << stras[1][0] in CheckStrategy()71 if (stras[0][1] != stras[1][1] || stras[0][1] != stras[2][1] || stras[0][1] != stras[3][0]) { in CheckStrategy()72 MS_LOG(ERROR) << name_ << ": Invalid strategy. The strategys[0][1]:" << stras[0][1] in CheckStrategy()73 << ", strategys[1][1]:" << stras[1][1] << ", strategys[2][1]:" << stras[2][1] in CheckStrategy()74 << ", strategys[3][0]:" << stras[3][0] << " should be the same."; in CheckStrategy()[all …]
52 Strategys stras = strategy->GetInputDim(); in CheckStrategy() local53 if (stras.size() != DSD_MATMUL_INPUTS_SIZE) { in CheckStrategy()57 …if (stras[0].size() != DSD_MATMUL_STRATEGY_W_SIZE || stras[1].size() != DSD_MATMUL_STRATEGY_W_SIZE… in CheckStrategy()58 stras[2].size() != DSD_MATMUL_STRATEGY_V_SIZE) { in CheckStrategy()62 int64_t batch_size_shard_num = stras[0][0]; in CheckStrategy()63 int64_t num_heads_shard_num = stras[0][1]; in CheckStrategy()64 for (size_t i = 0; i < stras.size(); ++i) { in CheckStrategy()65 for (size_t j = 0; j < stras[i].size(); ++j) { in CheckStrategy()67 if (stras[i][j] != batch_size_shard_num) { in CheckStrategy()73 if (stras[i][j] != num_heads_shard_num) { in CheckStrategy()[all …]
71 Strategys stras = strategy->GetInputDim(); in CheckStrategy() local76 for (Dimensions stra : stras) { in CheckStrategy()83 if (stras[0][0] != 1) { in CheckStrategy()
528 Strategys stras; in PrepareStrategy() local529 stras.push_back(input0_partitions); in PrepareStrategy()530 stras.push_back(input1_partitions); in PrepareStrategy()531 (*sp) = std::make_shared<Strategy>(stage_id, stras); in PrepareStrategy()
114 Strategys stras = strategy->GetInputDim(); in CheckStrategy() local115 for (Dimensions stra : stras) { in CheckStrategy()
1026 Strategys stras(inputs_partitions); in PrepareStrategyBase() local1027 (*sp) = std::make_shared<Strategy>(stage_id, stras); in PrepareStrategyBase()