Skip to content

Commit

Permalink
Decrease input sizes in test_slice_permute021_fusion (#822)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: #822

This attempts to resolve the CircleCI OOM issue.

Reviewed By: muchulee8, kadeng

Differential Revision: D47362870

fbshipit-source-id: a64da980894cd8bbd4e71f2f62ed7196a0f97734
  • Loading branch information
aakhundov authored and facebook-github-bot committed Jul 11, 2023
1 parent 0c221d9 commit ecb705a
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tests/unittest/compiler/test_slice_permute021_fusion.py
Original file line number Diff line number Diff line change
Expand Up @@ -123,19 +123,19 @@ def test_slice_permute021_fusion(self):
self._test_slice_permute021_fusion(
N=2,
K=2,
slice_input_shape=[120, 1211, 1200],
slice_input_shape=[3, 4, 120],
slice_start_indices=[0, 0, 3],
slice_end_indices=[None, None, 1100],
slice_end_indices=[None, None, 110],
dims=(0, 2, 1),
test_name="slice_permute021",
dtype="float16",
)
self._test_slice_permute021_fusion(
N=2,
K=2,
slice_input_shape=[123, 1211, 1200],
slice_input_shape=[3, 121, 4],
slice_start_indices=[0, 5, 0],
slice_end_indices=[None, 1200, None],
slice_end_indices=[None, 115, None],
dims=(0, 2, 1),
test_name="slice_permute021",
dtype="float16",
Expand Down

0 comments on commit ecb705a

Please sign in to comment.