[Mlir-commits] [mlir] [Fix][mlir](convert-tensor-to-spirv) mlir-opt crashes at BuiltinAttributes.cpp:366 with assertion (getType().isIndex() || getType().isSignlessInteger()) && 'must be signless integer' failed. (PR #204937)

[Chirag Wattamwar]

https://github.com/ChiragSW updated https://github.com/llvm/llvm-project/pull/204937

>From 5752d6bd6cf6240d6495102deef2208516f8cb99 Mon Sep 17 00:00:00 2001
From: Chirag <wattamwarchiraggg at gmail.com>
Date: Sat, 20 Jun 2026 22:30:43 +0530
Subject: [PATCH] [Fix][mlir](convert-tensor-to-spirv) mlir-opt crashes at
 BuiltinAttributes.cpp:366 with assertion (getType().isIndex() ||
 getType().isSignlessInteger()) && 'must be signless integer' failed.

---
 mlir/lib/Conversion/ArithToSPIRV/ArithToSPIRV.cpp  | 14 ++++++++++----
 .../TensorToSPIRV/tensor-ops-to-spirv.mlir         |  7 +++++++
 2 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/mlir/lib/Conversion/ArithToSPIRV/ArithToSPIRV.cpp b/mlir/lib/Conversion/ArithToSPIRV/ArithToSPIRV.cpp
index 9a6d330db72fe..ff524b5360670 100644
--- a/mlir/lib/Conversion/ArithToSPIRV/ArithToSPIRV.cpp
+++ b/mlir/lib/Conversion/ArithToSPIRV/ArithToSPIRV.cpp
@@ -53,18 +53,24 @@ static BoolAttr convertBoolAttr(Attribute srcAttr, Builder builder) {
 /// Returns null attribute if conversion fails.
 static IntegerAttr convertIntegerAttr(IntegerAttr srcAttr, IntegerType dstType,
                                       Builder builder) {
+  unsigned dstWidth = dstType.getWidth();
+  APInt srcValue = srcAttr.getValue();
+  bool isUnsigned = srcAttr.getType().isUnsignedInteger();
+  APInt dstValue = isUnsigned ? srcValue.zextOrTrunc(dstWidth)
+                              : srcValue.sextOrTrunc(dstWidth);
+
   // If the source number uses less active bits than the target bitwidth, then
   // it should be safe to convert.
-  if (srcAttr.getValue().isIntN(dstType.getWidth()))
-    return builder.getIntegerAttr(dstType, srcAttr.getInt());
+  if (srcValue.isIntN(dstWidth))
+    return builder.getIntegerAttr(dstType, dstValue);
 
   // XXX: Try again by interpreting the source number as a signed value.
   // Although integers in the standard dialect are signless, they can represent
   // a signed number. It's the operation decides how to interpret. This is
   // dangerous, but it seems there is no good way of handling this if we still
   // want to change the bitwidth. Emit a message at least.
-  if (srcAttr.getValue().isSignedIntN(dstType.getWidth())) {
-    auto dstAttr = builder.getIntegerAttr(dstType, srcAttr.getInt());
+  if (!isUnsigned && srcValue.isSignedIntN(dstWidth)) {
+    auto dstAttr = builder.getIntegerAttr(dstType, dstValue);
     LLVM_DEBUG(llvm::dbgs() << "attribute '" << srcAttr << "' converted to '"
                             << dstAttr << "' for type '" << dstType << "'\n");
     return dstAttr;
diff --git a/mlir/test/Conversion/TensorToSPIRV/tensor-ops-to-spirv.mlir b/mlir/test/Conversion/TensorToSPIRV/tensor-ops-to-spirv.mlir
index 65c6e0587129e..53ee699024068 100644
--- a/mlir/test/Conversion/TensorToSPIRV/tensor-ops-to-spirv.mlir
+++ b/mlir/test/Conversion/TensorToSPIRV/tensor-ops-to-spirv.mlir
@@ -72,6 +72,13 @@ func.func @tensor_2d() -> () {
   return
 }
 
+// CHECK-LABEL: func @tensor_unsigned_int
+// CHECK-NEXT:    spirv.Constant dense<[10, 200]> : tensor<2xui32> : !spirv.array<2 x ui32>
+func.func @tensor_unsigned_int() -> () {
+  %x = arith.constant dense<[10, 200]> : tensor<2xui8>
+  return
+}
+
 // We do not handle zero-element tensors yet. Just make we do not crash on them.
 // CHECK-LABEL: func @tensor_2d_empty
 // CHECK-NEXT:    arith.constant dense<>