未验证 提交 897fecf2 编写于 作者: K kangguangli 提交者: GitHub

fix typo in unit test for gpu kernel of transfer_layout (#46839)

上级 6c2af7bb
......@@ -73,7 +73,7 @@ class TestTransferLayoutOpGpu(unittest.TestCase):
dtype='float32',
name='x')
y = softmax_with_data_format(x, data_format='NCHW')
z = softmax_with_data_format(x, data_format='NHWC')
z = softmax_with_data_format(y, data_format='NHWC')
place = fluid.CUDAPlace(
0) if core.is_compiled_with_cuda() else fluid.CPUPlace()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册