From 70aed1e4fbea013358ce0ccc5129b8cb47cb8605 Mon Sep 17 00:00:00 2001 From: Adnan Akhundov Date: Tue, 11 Jul 2023 02:50:07 -0700 Subject: [PATCH] Decrease input sizes in test_slice_permute021_fusion Summary: This attempts to resolve the CircleCI OOM issue. Reviewed By: kadeng Differential Revision: D47362870 fbshipit-source-id: a81c7a24d613ba9d95d0e929a3cdbb32ba21cc3a --- tests/unittest/compiler/test_slice_permute021_fusion.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/unittest/compiler/test_slice_permute021_fusion.py b/tests/unittest/compiler/test_slice_permute021_fusion.py index ea6d57c0c..d7e8b40c8 100644 --- a/tests/unittest/compiler/test_slice_permute021_fusion.py +++ b/tests/unittest/compiler/test_slice_permute021_fusion.py @@ -123,9 +123,9 @@ 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", @@ -133,9 +133,9 @@ def test_slice_permute021_fusion(self): 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",