From f97796a66a922df14f2c1dab388797fe61ea09f1 Mon Sep 17 00:00:00 2001 From: wooway777 Date: Wed, 17 Dec 2025 13:17:48 +0800 Subject: [PATCH] issue/795 - add a test case for rearrange --- test/infiniop/rearrange.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/test/infiniop/rearrange.py b/test/infiniop/rearrange.py index 41d987a95..2e8f8d459 100644 --- a/test/infiniop/rearrange.py +++ b/test/infiniop/rearrange.py @@ -76,9 +76,18 @@ def column_major_strides(shape): column_major_strides((3, 4, 50, 50, 5, 7)), # y_stride ), ((15, 10752), (0, 1), (10752, 1)), - ((2, 2, 2, 2, 2, 2), (4, 8, 16, 32, 64, 128), (64, 32, 16, 8, 4, 2)), # shape # x_stride # y_stride + ( + (2, 2, 2, 2, 2, 2), + (4, 8, 16, 32, 64, 128), + (64, 32, 16, 8, 4, 2), + ), # shape # x_stride # y_stride ((8, 4, 20, 64), (5120, 64, 256, 1), None), # shape # x_stride # y_stride - ((8, 4, 20, 64), (5120, 64, 256, 1), (1048576, 262144, 64, 1)), # shape # x_stride # y_stride + ( + (8, 4, 20, 64), + (5120, 64, 256, 1), + (1048576, 262144, 64, 1), + ), # shape # x_stride # y_stride + ((13, 4), (10, 1), (10, 1)), # shape # x_stride # y_stride ] # Data types used for testing