mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!27538 check bprop shape change 

Merge pull request !27538 from caifubi/master-pynative-mindrt-fix-case
This commit is contained in:
i-robot 2021-12-13 01:31:18 +00:00 committed by Gitee
parent 8e4935f87c e722090e46
commit 34198dd525
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/st/pynative/test_pynative_hook.py
View File

@ -203,12 +203,13 @@ def test_pynative_custom_bprop_and_Cell_MulAdd():
@pytest.mark.platform_x86_ascend_training
@pytest.mark.env_onecard
def test_pynative_custom_bprop_and_Cell_Ms_Cell_Change_Shape():
context.set_context(check_bprop=True)
custom_cell = test_custom_cell_base()
ms_Cell = custom_cell.test_custom_cell_function(Ms_Cell_Change_Shape())
ms_Cell.bprop_debug = True
with pytest.raises(RuntimeError) as ex:
with pytest.raises(ValueError) as ex:
grad_all(ms_Cell)(Tensor(1, mstype.float32))
assert "Shapes of input and parameter are different, input index" in str(ex.value)
assert "should have the same shape as the 0th arg" in str(ex.value)
@pytest.mark.level1