[MLIR] Handle llvm.icmp of pointers

[rriddle]

Thanks! Is anything needed for import/export?

mlir/test/Dialect/LLVMIR/roundtrip.mlir

27

Can you add a pointer vector test?

[wsmoses]

mlir/test/Dialect/LLVMIR/roundtrip.mlir

27

An initial look at doing so demonstrates that vectors do not presently support LLVM.pointer as arguments, making such a test impossible at the moment. This should likely be remedied (being a ~index type) but in a future PR.

[wsmoses]

mlir/test/Dialect/LLVMIR/roundtrip.mlir

27

Oh whoops, used the wrong vector type in the test when I ran that. Done.

mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td

def ICmpPredicate : I64EnumAttr<
      ICmpPredicateSGT, ICmpPredicateSGE, ICmpPredicateULT, ICmpPredicateULE,
      ICmpPredicateUGT, ICmpPredicateUGE]> {
   let cppNamespace = "::mlir::LLVM";
 }
 
 // Other integer operations.
 def LLVM_ICmpOp : LLVM_Op<"icmp", [NoSideEffect]> {
   let arguments = (ins ICmpPredicate:$predicate,
-                   LLVM_ScalarOrVectorOf<LLVM_AnyInteger>:$lhs,
-                   LLVM_ScalarOrVectorOf<LLVM_AnyInteger>:$rhs);
+                   AnyTypeOf<[LLVM_ScalarOrVectorOf<LLVM_AnyInteger>, LLVM_ScalarOrVectorOf<LLVM_AnyPointer>]>:$lhs,
+                   AnyTypeOf<[LLVM_ScalarOrVectorOf<LLVM_AnyInteger>, LLVM_ScalarOrVectorOf<LLVM_AnyPointer>]>:$rhs);
   let results = (outs LLVM_ScalarOrVectorOf<I1>:$res);
   let llvmBuilder = [{
     $res = builder.CreateICmp(getLLVMCmpPredicate($predicate), $lhs, $rhs);
   }];
   let builders = [
     OpBuilder<(ins "ICmpPredicate":$predicate, "Value":$lhs, "Value":$rhs),
     [{
       build($_builder, $_state, IntegerType::get(lhs.getType().getContext(), 1),

mlir/test/Dialect/LLVMIR/roundtrip.mlir

 // CHECK: {{.*}} = llvm.add %[[I32]], %[[I32]] : i32
 // CHECK: {{.*}} = llvm.sub %[[I32]], %[[I32]] : i32
 // CHECK: {{.*}} = llvm.mul %[[I32]], %[[I32]] : i32
 // CHECK: {{.*}} = llvm.udiv %[[I32]], %[[I32]] : i32
 // CHECK: {{.*}} = llvm.sdiv %[[I32]], %[[I32]] : i32
 // CHECK: {{.*}} = llvm.urem %[[I32]], %[[I32]] : i32
 // CHECK: {{.*}} = llvm.srem %[[I32]], %[[I32]] : i32
 // CHECK: {{.*}} = llvm.icmp "ne" %[[I32]], %[[I32]] : i32
+// CHECK: {{.*}} = llvm.icmp "ne" %[[I8PTR1]], %[[I8PTR1]] : !llvm.ptr<i8>
   %0 = llvm.add %arg0, %arg0 : i32
   %1 = llvm.sub %arg0, %arg0 : i32
   %2 = llvm.mul %arg0, %arg0 : i32
   %3 = llvm.udiv %arg0, %arg0 : i32
   %4 = llvm.sdiv %arg0, %arg0 : i32
   %5 = llvm.urem %arg0, %arg0 : i32
   %6 = llvm.srem %arg0, %arg0 : i32
   %7 = llvm.icmp "ne" %arg0, %arg0 : i32
+  %ptrcmp = llvm.icmp "ne" %arg2, %arg2 : !llvm.ptr<i8>
 
 // Floating point binary operations.
 //
 // CHECK: {{.*}} = llvm.fadd %[[FLOAT]], %[[FLOAT]] : f32
 // CHECK: {{.*}} = llvm.fsub %[[FLOAT]], %[[FLOAT]] : f32
 // CHECK: {{.*}} = llvm.fmul %[[FLOAT]], %[[FLOAT]] : f32
 // CHECK: {{.*}} = llvm.fdiv %[[FLOAT]], %[[FLOAT]] : f32
 // CHECK: {{.*}} = llvm.frem %[[FLOAT]], %[[FLOAT]] : f32