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

!48615 modify cpu core dump 

Merge pull request !48615 from 王程浩/master
This commit is contained in:
i-robot 2023-02-09 11:14:11 +00:00 committed by Gitee
parent 49eeb3104e 2b4ed31bdc
commit d475c95bdf
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mindspore/ccsrc/plugin/device/cpu/kernel/sparse_segment_sqrt_n_cpu_kernel.cc
View File

@ -55,12 +55,22 @@ int SparseSegmentSqrtNCpuKernelMod::Resize(const BaseOperatorPtr &base_operator,
indices_shape_ = inputs.at(kIndex1)->GetDeviceShapeAdaptively();
segment_ids_shape_ = inputs.at(kIndex2)->GetDeviceShapeAdaptively();
y_shape_ = outputs.at(kIndex0)->GetDeviceShapeAdaptively();
is_null_input_ = CHECK_SHAPE_NULL(x_shape_, kernel_name_, "x_shape_") ||
CHECK_SHAPE_NULL(indices_shape_, kernel_name_, "indices_shape_") ||
CHECK_SHAPE_NULL(segment_ids_shape_, kernel_name_, "segment_ids_shape_");
if (is_null_input_) {
return KRET_OK;
}
return KRET_OK;
}
bool SparseSegmentSqrtNCpuKernelMod::Launch(const std::vector<kernel::AddressPtr> &inputs,
const std::vector<kernel::AddressPtr> &workspace,
const std::vector<kernel::AddressPtr> &outputs) {
if (is_null_input_) {
return true;
}
if (dtype_ == kNumberTypeFloat16) {
if (dtype1_ == kNumberTypeInt32) {
if (dtype2_ == kNumberTypeInt32) {

1
mindspore/ccsrc/plugin/device/cpu/kernel/sparse_segment_sqrt_n_cpu_kernel.h
View File

@ -54,6 +54,7 @@ class SparseSegmentSqrtNCpuKernelMod : public NativeCpuKernelMod {
TypeId dtype_{kTypeUnknown};
TypeId dtype1_{kTypeUnknown};
TypeId dtype2_{kTypeUnknown};
bool is_null_input_{false};
};
} // namespace kernel
} // namespace mindspore