未验证 提交 b83e506b 编写于 作者: H houj04 提交者: GitHub

[XPU] fix ut: test_kldiv_loss_op_xpu, test_temporal_shift_op_xpu (#52258)

* fix test_kldiv_loss_op_xpu

* fix test_temporal_shift_op_xpu
上级 37bd7e78
...@@ -80,7 +80,7 @@ class XPUTestKLDivLossOp(XPUOpTestWrapper): ...@@ -80,7 +80,7 @@ class XPUTestKLDivLossOp(XPUOpTestWrapper):
['X'], ['X'],
'Loss', 'Loss',
no_grad_set={"Target"}, no_grad_set={"Target"},
check_eager=True, check_dygraph=True,
) )
def initTestCase(self): def initTestCase(self):
......
...@@ -83,10 +83,10 @@ class XPUTestTemporalShiftOp(XPUOpTestWrapper): ...@@ -83,10 +83,10 @@ class XPUTestTemporalShiftOp(XPUOpTestWrapper):
self.python_out_sig = ["Out"] self.python_out_sig = ["Out"]
def test_check_output(self): def test_check_output(self):
self.check_output(check_dygraph=True) self.check_output(check_dygraph=False)
def test_check_grad(self): def test_check_grad(self):
self.check_grad(['X'], 'Out', check_dygraph=True) self.check_grad(['X'], 'Out', check_dygraph=False)
def initTestCase(self): def initTestCase(self):
self.x_shape = (6, 4, 4, 4) self.x_shape = (6, 4, 4, 4)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册