[Mlir-commits] [mlir] [mlir][spirv] Add pattern matching for arith.index_cast index to i1 for ArithToSPIRV (PR #156031)

[Jakub Kuderski]

================
@@ -607,6 +607,30 @@ struct UIToFPI1Pattern final : public OpConversionPattern<arith::UIToFPOp> {
   }
 };
 
+//===----------------------------------------------------------------------===//
+// IndexCastOp
+//===----------------------------------------------------------------------===//
+
+// Converts arith.index_cast to spirv.INotEqual if the target type is i1.
+struct IndexCastIndexI1Pattern final
+    : public OpConversionPattern<arith::IndexCastOp> {
+  using OpConversionPattern::OpConversionPattern;
+
+  LogicalResult
+  matchAndRewrite(arith::IndexCastOp op, OpAdaptor adaptor,
+                  ConversionPatternRewriter &rewriter) const override {
+    Type srcType = adaptor.getIn().getType();
+    if (!op.getType().isInteger(1))
+      return failure();
+
+    Location loc = op.getLoc();
+    Value zeroIdx = spirv::ConstantOp::getZero(srcType, loc, rewriter);
----------------
kuhar wrote:

You can inline `srcType` here, it's not used anywhere else

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