From b896609e0ccf19c5412ad336736adb4637b906ae Mon Sep 17 00:00:00 2001 From: Cong Ma Date: Wed, 14 Aug 2024 20:16:34 +0000 Subject: [PATCH] Add test cases back to mode_test_params.yaml Add them back when CK's patch is merged. --- .../01_contraction/configs/mode_test_params.yaml | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/test/01_contraction/configs/mode_test_params.yaml b/test/01_contraction/configs/mode_test_params.yaml index 494c498d..16dbb9f0 100644 --- a/test/01_contraction/configs/mode_test_params.yaml +++ b/test/01_contraction/configs/mode_test_params.yaml @@ -26,13 +26,13 @@ Lengths: # M0K4 N0K4 M0N0 - [[3, 3], [2, 3], [3, 2]] # M0K0M1K1M2K2M3K3M4K4M5K5 N0K0N1K1N2K2N3K3N4K4N5K5 M0N0M1N1M2N2M3N3M4N4M5N5 - # - [[3, 1, 2, 2, 3, 4, 3, 5, 4, 3, 3, 5], [2, 1, 4, 2, 3, 4, 3, 5, 1, 3, 4, 5], [3, 2, 2, 4, 3, 3, 3, 3, 4, 1, 3, 4]] + - [[3, 1, 2, 2, 3, 4, 3, 5, 4, 3, 3, 5], [2, 1, 4, 2, 3, 4, 3, 5, 1, 3, 4, 5], [3, 2, 2, 4, 3, 3, 3, 3, 4, 1, 3, 4]] # M5M2K2K4K1K3K0M1K5M3M0M4 K1K3K0N5N1N0K2K5K4N2N4N3 N2M4N3N4M2M5M3N1M0M1N5N0 - # - [[3, 3, 4, 3, 2, 5, 1, 2, 5, 3, 3, 4], [2, 5, 1, 4, 4, 2, 4, 5, 3, 3, 1, 3], [3, 4, 3, 1, 3, 3, 3, 4, 3, 2, 4, 2]] + - [[3, 3, 4, 3, 2, 5, 1, 2, 5, 3, 3, 4], [2, 5, 1, 4, 4, 2, 4, 5, 3, 3, 1, 3], [3, 4, 3, 1, 3, 3, 3, 4, 3, 2, 4, 2]] # M0K2M3K5M2K4M5K3K0K1M1M4 M0M1M3K5K1K4M2K0M5M4K2K3 K1M0M2K4K5M4K0K2M1K3M5M3 - # - [[3, 4, 3, 5, 3, 3, 3, 5, 1, 2, 2, 4], [2, 4, 3, 5, 2, 3, 3, 1, 4, 1, 4, 5], [4, 3, 3, 1, 4, 4, 2, 3, 2, 3, 3, 3]] + - [[3, 4, 3, 5, 3, 3, 3, 5, 1, 2, 2, 4], [2, 4, 3, 5, 2, 3, 3, 1, 4, 1, 4, 5], [4, 3, 3, 1, 4, 4, 2, 3, 2, 3, 3, 3]] # K5K3M0M5K4K1M3K2M1K0M4M2 M4M0M3M5K4M2K2K3M1K5K1K0 K1K4K3M2M4K5M5M0K0M1M3K2 - # - [[5, 5, 3, 3, 3, 2, 3, 4, 2, 1, 4, 3], [1, 2, 3, 4, 3, 3, 4, 5, 4, 5, 2, 1], [4, 1, 3, 3, 4, 4, 3, 3, 2, 2, 3, 3]] + - [[5, 5, 3, 3, 3, 2, 3, 4, 2, 1, 4, 3], [1, 2, 3, 4, 3, 3, 4, 5, 4, 5, 2, 1], [4, 1, 3, 3, 4, 4, 3, 3, 2, 2, 3, 3]] Strides: - [] @@ -50,11 +50,11 @@ Modes: # M0K4 N0K4 M0N0 - [[0, 16], [6, 16], [0, 6]] # M0K0M1K1M2K2M3K3M4K4M5K5 N0K0N1K1N2K2N3K3N4K4N5K5 M0N0M1N1M2N2M3N3M4N4M5N5 - # - [[0, 12, 1, 13, 2, 14, 3, 15, 4, 16, 5, 17], [6, 12, 7, 13, 8, 14, 9, 15, 10, 16, 11, 17], [0, 6, 1, 7, 2, 8, 3, 9, 4, 10, 5, 11]] + - [[0, 12, 1, 13, 2, 14, 3, 15, 4, 16, 5, 17], [6, 12, 7, 13, 8, 14, 9, 15, 10, 16, 11, 17], [0, 6, 1, 7, 2, 8, 3, 9, 4, 10, 5, 11]] # M5M2K2K4K1K3K0M1K5M3M0M4 K1K3K0N5N1N0K2K5K4N2N4N3 N2M4N3N4M2M5M3N1M0M1N5N0 - # - [[5, 2, 14, 16, 13, 15, 12, 1, 17, 3, 0, 4], [13, 15, 12, 11, 7, 6, 14, 17, 16, 8, 10, 9], [8, 4, 9, 10, 2, 5, 3, 7, 0, 1, 11, 6]] + - [[5, 2, 14, 16, 13, 15, 12, 1, 17, 3, 0, 4], [13, 15, 12, 11, 7, 6, 14, 17, 16, 8, 10, 9], [8, 4, 9, 10, 2, 5, 3, 7, 0, 1, 11, 6]] # M0K2M3K5M2K4M5K3K0K1M1M4 M0M1M3K5K1K4M2K0M5M4K2K3 K1M0M2K4K5M4K0K2M1K3M5M3 - # - [[0, 14, 3, 17, 2, 16, 5, 15, 12, 13, 1, 4], [6, 7, 9, 17, 13, 16, 8, 12, 11, 10, 14, 15], [7, 0, 2, 10, 11, 4, 6, 8, 1, 9, 5, 3]] + - [[0, 14, 3, 17, 2, 16, 5, 15, 12, 13, 1, 4], [6, 7, 9, 17, 13, 16, 8, 12, 11, 10, 14, 15], [7, 0, 2, 10, 11, 4, 6, 8, 1, 9, 5, 3]] # K5K3M0M5K4K1M3K2M1K0M4M2 M4M0M3M5K4M2K2K3M1K5K1K0 K1K4K3M2M4K5M5M0K0M1M3K2 - # - [[17, 15, 0, 5, 16, 13, 3, 14, 1, 12, 4, 2], [10, 6, 9, 11, 16, 8, 14, 15, 7, 17, 13, 12], [7, 10, 9, 2, 4, 11, 5, 0, 6, 1, 3, 8]] + - [[17, 15, 0, 5, 16, 13, 3, 14, 1, 12, 4, 2], [10, 6, 9, 11, 16, 8, 14, 15, 7, 17, 13, 12], [7, 10, 9, 2, 4, 11, 5, 0, 6, 1, 3, 8]] ...