Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
* Fix test case gt
Browse files Browse the repository at this point in the history
  • Loading branch information
@hanke580
hanke580 committed Jun 3, 2020
1 parent d4a55ad commit 3802104
Showing 1 changed file with 12 additions and 12 deletions.
24 changes: 12 additions & 12 deletions tests/python/unittest/test_numpy_op.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -7817,25 +7817,25 @@ def dbg(name, data):
('ij, ij -> i', [(1, 4), (2, 4)], lambda *args: (_np.sum(args[1], axis=0)[None, :],
_np.tile(args[0], [2, 1]))),
# one dimensim bug
('...ij, ...jk -> ...ik', [(1, 4), (4, 2)], lambda *args: (np.sum(args[1], axis=1)[None, :],
('...ij, ...jk -> ...ik', [(1, 4), (4, 2)], lambda *args: (args[1].sum(axis=1)[None, :],
_np.tile(args[0].sum(axis=0)[: ,None], [1, 2]))),
('...ij, ...jk -> ...ik', [(2, 4), (4, 2)], lambda *args: (_np.tile(args[1].sum(axis=1)[None, :], [2, 1]),
_np.tile(args[0].sum(axis=0)[: ,None], [1, 2]))),
('...ij, ...jk -> ...ik', [(3, 2, 1, 4), (3, 2, 4, 2)], lambda *args: (
args[1].sum(axis=3)[:, :, None, :],
_np.tile(args[0].sum(axis=2)[:, :, :, None], [1, 1, 1, 2]))),
args[1].sum(axis=3)[:, :, None, :],
_np.tile(args[0].sum(axis=2)[:, :, :, None], [1, 1, 1, 2]))),
('...ij, ...ik -> ...jk', [(1, 1, 1, 4), (1, 1, 1, 3)], lambda *args: (
_np.tile(args[1].sum(axis=3)[:, :, :, None], [1, 1, 1, 4]),
_np.tile(args[0].sum(axis=3)[:, :, : ,None], [1, 1, 1, 3]))),
_np.tile(args[1].sum(axis=3)[:, :, :, None], [1, 1, 1, 4]),
_np.tile(args[0].sum(axis=3)[:, :, : ,None], [1, 1, 1, 3]))),
('...ij, ...jc -> ...ic', [(1, 1, 5, 3), (1, 1, 3, 2)], lambda *args: (
_np.tile(args[1].sum(axis=3)[:, :, None, :], [1, 1, 5, 1]),
_np.tile(args[0].sum(axis=2)[:, :, : ,None], [1, 1, 1, 2]))),
('...ij, ...jc -> ...ic', [(1, 2, 5, 4), (1, 1, 4, 2)], lambda *args: (
_np.tile(args[1].sum(axis=3)[:, :, None, :], [1, 1, 5, 1]),
_np.tile(args[0].sum(axis=2)[:, :, : ,None], [1, 1, 1, 2]))),
_np.tile(args[1].sum(axis=3)[:, :, None, :], [1, 1, 5, 1]),
_np.tile(args[0].sum(axis=2)[:, :, : ,None], [1, 1, 1, 2]))),
('...ij, ...jc -> ...ic', [(1, 2, 5, 4), (1, 2, 4, 2)], lambda *args: (
_np.tile(args[1].sum(axis=3)[:, :, None, :], [1, 1, 5, 1]),
_np.tile(args[0].sum(axis=2)[:, :, : ,None], [1, 1, 1, 2]))),
('...ij, ...jc -> ...ic', [(2, 1, 5, 4), (2, 1, 4, 2)], lambda *args: (
_np.tile(args[1].sum(axis=3)[:, :, None, :], [1, 1, 5, 1]),
_np.tile(args[0].sum(axis=2)[:, :, : ,None], [1, 1, 1, 2]))),
_np.tile(args[1].sum(axis=3)[:, :, None, :], [1, 1, 5, 1]),
_np.tile(args[0].sum(axis=2)[:, :, : ,None], [1, 1, 1, 2]))),
# issue #16576
# commented due to long running time
# ('abiz,abjz->abij', [(64, 8, 128, 512), (64, 8, 128, 512)], lambda *args: (_np.matmul(_np.ones((64, 8, 128, 128)), args[1]),
Expand Down

0 comments on commit 3802104

Please sign in to comment.