Method strided_slice

tests/python/relax/test_op_index.py:1002–1011

(
            A: T.Buffer((T.int64(16), T.int64(16))),
            B: T.Buffer((T.int64(1), T.int64(16))),
            index: T.int64,
        )

        @T.prim_func(private=True, s_tir=True)
        def strided_slice(
            A: T.Buffer((T.int64(16), T.int64(16))),
            B: T.Buffer((T.int64(1), T.int64(16))),
            index: T.int64,
        ):
            T.func_attr({"tirx.noalias": True})
            for iters in T.grid(*B.shape):
                with T.sblock("T_dynamic_strided_slice"):
                    i, j = T.axis.remap("SS", iters)
                    B[i, j] = A[i + index, j]

    after = tvm.relax.transform.LegalizeOps()(before)
    tvm.ir.assert_structural_equal(expected, after)