(
rxplaceholder: T.Buffer((T.int64(1), T.int64(4), T.int64(64), T.int64(64)), "float32"),
rxplaceholder_1: T.Buffer(
(T.int64(1), T.int64(4), T.int64(64), T.int64(64)), "float32"
),
T_concat: T.Buffer((T.int64(2), T.int64(4), T.int64(64), T.int64(64)), "float32"),
)
| 1047 | class Module: |
| 1048 | @T.prim_func(private=True, s_tir=True) |
| 1049 | def concatenate( |
| 1050 | rxplaceholder: T.Buffer((T.int64(1), T.int64(4), T.int64(64), T.int64(64)), "float32"), |
| 1051 | rxplaceholder_1: T.Buffer( |
| 1052 | (T.int64(1), T.int64(4), T.int64(64), T.int64(64)), "float32" |
| 1053 | ), |
| 1054 | T_concat: T.Buffer((T.int64(2), T.int64(4), T.int64(64), T.int64(64)), "float32"), |
| 1055 | ): |
| 1056 | T.func_attr({"op_pattern": 2, "tirx.noalias": True}) |
| 1057 | for ax0, ax1, ax2, ax3 in T.grid(T.int64(2), T.int64(4), T.int64(64), T.int64(64)): |
| 1058 | with T.sblock("T_concat"): |
| 1059 | v_ax0, v_ax1, v_ax2, v_ax3 = T.axis.remap("SSSS", [ax0, ax1, ax2, ax3]) |
| 1060 | T.reads( |
| 1061 | rxplaceholder_1[v_ax0 - T.int64(1), v_ax1, v_ax2, v_ax3], |
| 1062 | rxplaceholder[v_ax0, v_ax1, v_ax2, v_ax3], |
| 1063 | ) |
| 1064 | T.writes(T_concat[v_ax0, v_ax1, v_ax2, v_ax3]) |
| 1065 | T_concat[v_ax0, v_ax1, v_ax2, v_ax3] = T.if_then_else( |
| 1066 | T.int64(1) <= v_ax0, |
| 1067 | rxplaceholder_1[v_ax0 - T.int64(1), v_ax1, v_ax2, v_ax3], |
| 1068 | rxplaceholder[v_ax0, v_ax1, v_ax2, v_ax3], |
| 1069 | ) |
| 1070 | |
| 1071 | @T.prim_func(private=True, s_tir=True) |
| 1072 | def transpose2( |
no test coverage detected