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

!31758 restore testcase bug of broadcastto op 

Merge pull request !31758 from wangyanling/master
This commit is contained in:
i-robot 2022-03-24 08:03:23 +00:00 committed by Gitee
parent 56ca83a1ec 7ce37d2990
commit 4f480138d5
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/plugin/device/cpu/kernel/broadcast_to_cpu_kernel.cc
View File

@ -141,7 +141,7 @@ bool BroadcastToCpuKernelMod::LaunchKernel(const std::vector<AddressPtr> &inputs
std::vector<KernelAttr> BroadcastToCpuKernelMod::GetOpSupport() {
auto iter = func_list_.find(kernel_type_);
if (iter == func_list_.end()) {
MS_LOG(EXCEPTION) << "Does not support " << kernel_type_ << "!";
MS_LOG(EXCEPTION) << "not support " << kernel_type_ << "!";
}
std::vector<KernelAttr> support_list;

2
tests/st/ops/cpu/test_broadcast_to_op.py
View File

@ -115,5 +115,5 @@ def test_broadcast_dyn_invalid_init():
context.set_context(mode=context.GRAPH_MODE, device_target='CPU')
ms_shape = (2, -1, 4, 5)
x_np = np.random.rand(4, 5).astype(np.float32)
with pytest.raises(ValueError):
with pytest.raises(ValueError) or pytest.raises(RuntimeError):
P.BroadcastTo(ms_shape)(Tensor(x_np))