xref: /llvm-project/mlir/test/Dialect/Index/index-ops.mlir (revision 8cdb4f9f66439729ff7a6781fd7d765f9e2b44c4)

// RUN: mlir-opt %s | mlir-opt | FileCheck %s

// CHECK-LABEL: @binary_ops
// CHECK-SAME: %[[A:.*]]: index, %[[B:.*]]: index
func.func @binary_ops(%a: index, %b: index) {
  // CHECK-NEXT: index.add %[[A]], %[[B]]
  %0 = index.add %a, %b
  // CHECK-NEXT: index.sub %[[A]], %[[B]]
  %1 = index.sub %a, %b
  // CHECK-NEXT: index.mul %[[A]], %[[B]]
  %2 = index.mul %a, %b
  // CHECK-NEXT: index.divs %[[A]], %[[B]]
  %3 = index.divs %a, %b
  // CHECK-NEXT: index.divu %[[A]], %[[B]]
  %4 = index.divu %a, %b
  // CHECK-NEXT: index.ceildivs %[[A]], %[[B]]
  %5 = index.ceildivs %a, %b
  // CHECK-NEXT: index.ceildivu %[[A]], %[[B]]
  %6 = index.ceildivu %a, %b
  // CHECK-NEXT: index.floordivs %[[A]], %[[B]]
  %7 = index.floordivs %a, %b
  // CHECK-NEXT: index.rems %[[A]], %[[B]]
  %8 = index.rems %a, %b
  // CHECK-NEXT: index.remu %[[A]], %[[B]]
  %9 = index.remu %a, %b
  // CHECK-NEXT: index.maxs %[[A]], %[[B]]
  %10 = index.maxs %a, %b
  // CHECK-NEXT: index.maxu %[[A]], %[[B]]
  %11 = index.maxu %a, %b
  // CHECK-NEXT: index.mins %[[A]], %[[B]]
  %12 = index.mins %a, %b
  // CHECK-NEXT: index.minu %[[A]], %[[B]]
  %13 = index.minu %a, %b
  // CHECK-NEXT: index.shl %[[A]], %[[B]]
  %14 = index.shl %a, %b
  // CHECK-NEXT: index.shrs %[[A]], %[[B]]
  %15 = index.shrs %a, %b
  // CHECK-NEXT: index.shru %[[A]], %[[B]]
  %16 = index.shru %a, %b
  // CHECK-NEXT: index.and %[[A]], %[[B]]
  %17 = index.and %a, %b
  // CHECK-NEXT: index.or %[[A]], %[[B]]
  %18 = index.or %a, %b
  // CHECK-NEXT: index.xor %[[A]], %[[B]]
  %19 = index.xor %a, %b
  return
}

// CHECK-LABEL: @cmp_op
// CHECK-SAME: %[[A:.*]]: index, %[[B:.*]]: index
func.func @cmp_op(%a: index, %b: index) {
  // CHECK-NEXT: index.cmp eq(%[[A]], %[[B]])
  %0 = index.cmp eq(%a, %b)
  // CHECK-NEXT: index.cmp ne(%[[A]], %[[B]])
  %1 = index.cmp ne(%a, %b)
  // CHECK-NEXT: index.cmp slt(%[[A]], %[[B]])
  %2 = index.cmp slt(%a, %b)
  // CHECK-NEXT: index.cmp sle(%[[A]], %[[B]])
  %3 = index.cmp sle(%a, %b)
  // CHECK-NEXT: index.cmp sgt(%[[A]], %[[B]])
  %4 = index.cmp sgt(%a, %b)
  // CHECK-NEXT: index.cmp sge(%[[A]], %[[B]])
  %5 = index.cmp sge(%a, %b)
  // CHECK-NEXT: index.cmp ult(%[[A]], %[[B]])
  %6 = index.cmp ult(%a, %b)
  // CHECK-NEXT: index.cmp ule(%[[A]], %[[B]])
  %7 = index.cmp ule(%a, %b)
  // CHECK-NEXT: index.cmp ugt(%[[A]], %[[B]])
  %8 = index.cmp ugt(%a, %b)
  // CHECK-NEXT: index.cmp uge(%[[A]], %[[B]])
  %9 = index.cmp uge(%a, %b)
  return
}

// CHECK-LABEL: @sizeof_op
func.func @sizeof_op() {
  // CHECK: index.sizeof
  %0 = index.sizeof
  return
}

// CHECK-LABEL: @constant_op
func.func @constant_op() {
  // CHECK-NEXT: %idx0 = index.constant 0
  %0 = index.constant 0
  // CHECK-NEXT: %idx1 = index.constant 1
  %1 = index.constant 1
  // CHECK-NEXT: %idx42 = index.constant 42
  %2 = index.constant 42
  return
}

// CHECK-LABEL: @bool_constant_op
func.func @bool_constant_op() {
  // CHECK-NEXT: %true = index.bool.constant true
  %0 = index.bool.constant true
  // CHECK-NEXT: %false = index.bool.constant false
  %1 = index.bool.constant false
  return
}

// CHECK-LABEL: @cast_op
// CHECK-SAME: %[[A:.*]]: index, %[[B:.*]]: i32, %[[C:.*]]: i64
func.func @cast_op(%a: index, %b: i32, %c: i64) {
  // CHECK-NEXT: index.casts %[[A]] : index to i64
  %0 = index.casts %a : index to i64
  // CHECK-NEXT: index.casts %[[B]] : i32 to index
  %1 = index.casts %b : i32 to index
  // CHECK-NEXT: index.casts %[[C]] : i64 to index
  %2 = index.casts %c : i64 to index
  // CHECK-NEXT: index.castu %[[A]] : index to i64
  %3 = index.castu %a : index to i64
  // CHECK-NEXT: index.castu %[[B]] : i32 to index
  %4 = index.castu %b : i32 to index
  // CHECK-NEXT: index.castu %[[C]] : i64 to index
  %5 = index.castu %c : i64 to index
  return
}