Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!33108 upgrade Ascend package 15 Apr 22 on master 

Merge pull request !33108 from shenwei41/upgrade_ascend_20220415_master
This commit is contained in:
i-robot 2022-04-16 18:31:42 +00:00 committed by Gitee
parent 2174f71616 4a973c05da
commit b0a6995822
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
3 changed files with 37 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
graphengine

@ -1 +1 @@
Subproject commit 1f9abf3951bda21eeda511b2502b2bd5159a4a13
Subproject commit d63e074e5882d0461769cc0893e7e722ff1695f7

34
mindspore/ccsrc/plugin/device/ascend/kernel/tbe/tbe_kernel_select/tbe_property_checker.cc
View File

@ -41,7 +41,41 @@ static bool CheckStridedSlice(const CNodePtr &cnode) {
if (!strides.empty() && strides[strides.size() - 1] != 1) {
return false;
}
} else {
auto inputs = cnode->inputs();
const int input_num = 5;
if (inputs.size() < input_num) {
MS_EXCEPTION(ArgumentError) << "StridedSliceGrad should have 5 inputs, but got: " << inputs.size();
}
auto input_node = inputs[input_num - 1];
MS_EXCEPTION_IF_NULL(input_node);
auto value_node = input_node->cast<ValueNodePtr>();
MS_EXCEPTION_IF_NULL(value_node);
auto value = value_node->value();
if (value->isa<tensor::Tensor>()) {
auto tensor = value->cast<tensor::TensorPtr>();
TypePtr data_type = tensor->Dtype();
MS_EXCEPTION_IF_NULL(data_type);
TypeId type_id = data_type->type_id();
auto element_size = tensor->data().size();
if (type_id == kNumberTypeInt32) {
auto *data = reinterpret_cast<int *>(tensor->data_c());
if ((data[element_size - 1]) != 1) {
return false;
}
} else if (type_id == kNumberTypeInt64) {
auto *data = reinterpret_cast<int64_t *>(tensor->data_c());
if ((data[element_size - 1]) != 1) {
return false;
}
} else {
MS_EXCEPTION(TypeError) << "The strides of StridedSliceGrad must be int.";
}
} else {
MS_EXCEPTION(ValueError) << "The strides of StridedSliceGrad must be a constant." << inputs.size();
}
}
// check reduction on the last dimension
if (GetCNodeFuncName(cnode) == kStridedSliceOpName && common::AnfAlgo::HasNodeAttr(kAttrShrinkAxisMask, cnode)) {
auto shrink_axis_mask = static_cast<int>(common::AnfAlgo::GetNodeAttr<int64_t>(cnode, kAttrShrinkAxisMask));

4
tests/st/networks/models/bert/bert_performance/test_bert_tdt_lossscale.py
View File

@ -244,12 +244,12 @@ def test_bert_performance():
assert np.allclose(loss_scale, expect_loss_scale, 0, 0)
epoch_mseconds = np.array(time_monitor_callback.epoch_mseconds_list)[2]
expect_epoch_mseconds = 1400
expect_epoch_mseconds = 1500
print("epoch mseconds: {}".format(epoch_mseconds))
assert epoch_mseconds <= expect_epoch_mseconds + 5
per_step_mseconds = np.array(time_monitor_callback.per_step_mseconds_list)[2]
expect_per_step_mseconds = 14
expect_per_step_mseconds = 15
print("per step mseconds: {}".format(per_step_mseconds))
assert per_step_mseconds <= expect_per_step_mseconds + 1