[Mlir-commits] [mlir] [mlir][tosa] Fix indexing in TosaToTensor (PR #140906)

Luke Hutton llvmlistbot at llvm.org
Thu May 22 02:53:00 PDT 2025


================
@@ -700,3 +700,25 @@ func.func @concat_non_axis_dyn_mixed(%arg0: tensor<?x1xf32>, %arg1: tensor<?x1xf
   %0 = "tosa.concat"(%arg0, %arg1, %arg2) <{axis = 1 : i32}> : (tensor<?x1xf32>, tensor<?x1xf32>, tensor<?x1xf32>) -> tensor<5x3xf32>
   return
 }
+
+// -----
+
+// CHECK-LABEL: func @pad_extraction
+// CHECK-SAME: (%[[ARG0_SSA:.*]]: tensor<2x2xi32>, %[[PAD_INPUT_TENSOR_SSA:.*]]: tensor<1xi32>)
+func.func @pad_extraction(%arg0: tensor<2x2xi32>, %pad_input_tensor: tensor<1xi32>) -> tensor<4x5xi32> {
----------------
lhutton1 wrote:

nit: perhaps `pad_variable_pad_const` would be a more descriptive test name?

https://github.com/llvm/llvm-project/pull/140906


More information about the Mlir-commits mailing list