This commit is contained in:
ligan 2022-09-29 11:01:50 +08:00
parent 8b6924fee4
commit bcf4a0c5e4
1 changed files with 1 additions and 1 deletions

View File

@ -62,7 +62,7 @@ def test_broadcast_grad_cpu_type():
input_y = np.arange(88, 2 * 3 * 2 + 88).reshape((2, 3, 2))
input_dout = np.maximum(input_x, input_y)
net = MaxmumGradNet()
dtypes = (np.int8, np.int16, np.int32, np.int64, np.float16,
dtypes = (np.int16, np.int32, np.int64, np.float16,
np.float32, np.float64, np.uint16, np.uint32, np.uint64)
for dtype in dtypes:
result = net(Tensor(input_x.astype(dtype)), Tensor(input_y.astype(dtype)),