未验证 提交 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):
['X'],
'Loss',
no_grad_set={"Target"},
check_eager=True,
check_dygraph=True,
)
def initTestCase(self):
......
......@@ -83,10 +83,10 @@ class XPUTestTemporalShiftOp(XPUOpTestWrapper):
self.python_out_sig = ["Out"]
def test_check_output(self):
self.check_output(check_dygraph=True)
self.check_output(check_dygraph=False)
def test_check_grad(self):
self.check_grad(['X'], 'Out', check_dygraph=True)
self.check_grad(['X'], 'Out', check_dygraph=False)
def initTestCase(self):
self.x_shape = (6, 4, 4, 4)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册