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

[Shay Kleiman]

https://github.com/shay-kl created https://github.com/llvm/llvm-project/pull/140903

Changed the indexing used in the extractOp from one that is intended for 0d tensors to one that is intended for 1d tensors.

>From 1f5cd5d4f030e5442e2a9c8835b85e64d29028da Mon Sep 17 00:00:00 2001
From: Shay Kleiman <shay.kleiman at mobileye.com>
Date: Wed, 21 May 2025 16:59:03 +0300
Subject: [PATCH] [mlir][tosa] Fix indexing in TosaToTensor

Changed the indexing used in the extractOp from one that is intended for
0d tensors to one that is intended for 1d tensors.
---
 mlir/lib/Conversion/TosaToTensor/TosaToTensor.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mlir/lib/Conversion/TosaToTensor/TosaToTensor.cpp b/mlir/lib/Conversion/TosaToTensor/TosaToTensor.cpp
index 5f23a33049f87..6c68a7fc25103 100644
--- a/mlir/lib/Conversion/TosaToTensor/TosaToTensor.cpp
+++ b/mlir/lib/Conversion/TosaToTensor/TosaToTensor.cpp
@@ -362,7 +362,7 @@ class PadConverter : public OpConversionPattern<tosa::PadOp> {
     // Setup the default constantAttr.
 
     Value padConstant = rewriter.createOrFold<tensor::ExtractOp>(
-        loc, padOp.getPadConst(), ValueRange({}));
+      loc, padOp.getPadConst(), ValueRange({rewriter.create<arith::ConstantIndexOp>(loc, 0)}));
 
     if (!padConstant) {
       return rewriter.notifyMatchFailure(