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

change test level

This commit is contained in:
VectorSL 2021-10-30 10:19:18 +08:00
parent 289986fdfa
commit 6d2704b227
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/backend/kernel_compiler/gpu/other/dynamic_stitch_gpu_kernel.cc
View File

@ -44,7 +44,7 @@ bool DynamicStitchKernel::Init(const CNodePtr &kernel_node) {
auto data_type = AnfAlgo::GetInputDeviceDataType(kernel_node, n_);
// Index type is restricted to int32 by kernel prim.
size_t index_type_size = sizeof(int);
data_type_size_ = GetDtypeNbyte(TypeIdToString(data_type, false));
data_type_size_ = GetDtypeNbyte(TypeIdToString(data_type, true));
auto first_data_shape = AnfAlgo::GetInputRealDeviceShapeIfExist(kernel_node, n_);
one_data_ele_num_ = first_data_shape[first_data_shape.size() - 1];
for (size_t i = 0; i < n_; i++) {

2
tests/st/ops/gpu/test_dynamic_stitch_op.py
View File

@ -32,7 +32,7 @@ class Net(nn.Cell):
def construct(self, indices, data):
return self.stitch(indices, data)
@pytest.mark.level1
@pytest.mark.level0
@pytest.mark.platform_x86_gpu_training
@pytest.mark.env_onecard
def test_net_int32():