[Mlir-commits] [mlir] 883048b - [mlir][arith] Add integration test for shrui emulation

[Jakub Kuderski]

Author: Jakub Kuderski
Date: 2022-09-20T11:29:05-04:00
New Revision: 883048be84779e730b09d0826607d6cac45fad03

URL: https://github.com/llvm/llvm-project/commit/883048be84779e730b09d0826607d6cac45fad03
DIFF: https://github.com/llvm/llvm-project/commit/883048be84779e730b09d0826607d6cac45fad03.diff

LOG: [mlir][arith] Add integration test for shrui emulation

The new test cases focus on known edge cases in the current implementation.
Specifically, we check for low (0, 1), mid (7, 8, 9) and high (15) shift amounts with i16 operands.

Reviewed By: antiagainst

Differential Revision: https://reviews.llvm.org/D134182

Added: 
    mlir/test/Integration/Dialect/Arithmetic/CPU/test-wide-int-emulation-shrui-i16.mlir

Modified: 
    

Removed: 
    


################################################################################
diff  --git a/mlir/test/Integration/Dialect/Arithmetic/CPU/test-wide-int-emulation-shrui-i16.mlir b/mlir/test/Integration/Dialect/Arithmetic/CPU/test-wide-int-emulation-shrui-i16.mlir
new file mode 100644
index 0000000000000..5663723d4267c
--- /dev/null
+++ b/mlir/test/Integration/Dialect/Arithmetic/CPU/test-wide-int-emulation-shrui-i16.mlir
@@ -0,0 +1,85 @@
+// Check that the wide integer `arith.shrui` emulation produces the same result as wide
+// `arith.shrui`. Emulate i16 ops with i8 ops.
+
+// RUN: mlir-opt %s --convert-scf-to-cf --convert-cf-to-llvm --convert-vector-to-llvm \
+// RUN:             --convert-func-to-llvm --convert-arith-to-llvm | \
+// RUN:   mlir-cpu-runner -e entry -entry-point-result=void \
+// RUN:                   --shared-libs=%mlir_lib_dir/libmlir_c_runner_utils%shlibext | \
+// RUN:   FileCheck %s --match-full-lines
+
+// RUN: mlir-opt %s --test-arith-emulate-wide-int="widest-int-supported=8" \
+// RUN:             --convert-scf-to-cf --convert-cf-to-llvm --convert-vector-to-llvm \
+// RUN:             --convert-func-to-llvm --convert-arith-to-llvm | \
+// RUN:   mlir-cpu-runner -e entry -entry-point-result=void \
+// RUN:                   --shared-libs=%mlir_lib_dir/libmlir_c_runner_utils%shlibext | \
+// RUN:   FileCheck %s --match-full-lines
+
+// Ops in this function *only* will be emulated using i8 types.
+func.func @emulate_shrui(%lhs : i16, %rhs : i16) -> (i16) {
+  %res = arith.shrui %lhs, %rhs : i16
+  return %res : i16
+}
+
+func.func @check_shrui(%lhs : i16, %rhs : i16) -> () {
+  %res = func.call @emulate_shrui(%lhs, %rhs) : (i16, i16) -> (i16)
+  vector.print %res : i16
+  return
+}
+
+func.func @entry() {
+  %cst0 = arith.constant 0 : i16
+  %cst1 = arith.constant 1 : i16
+  %cst2 = arith.constant 2 : i16
+  %cst7 = arith.constant 7 : i16
+  %cst8 = arith.constant 8 : i16
+  %cst9 = arith.constant 9 : i16
+  %cst15 = arith.constant 15 : i16
+
+  %cst_1 = arith.constant -1 : i16
+  %cst_3 = arith.constant -3 : i16
+
+  %cst1337 = arith.constant 1337 : i16
+
+  %cst_i16_min = arith.constant -32768 : i16
+
+  // CHECK:      0
+  // CHECK-NEXT: 0
+  // CHECK-NEXT: 0
+  // CHECK-NEXT: 1
+  // CHECK-NEXT: 32767
+  // CHECK-NEXT: 1
+  func.call @check_shrui(%cst0, %cst0) : (i16, i16) -> ()
+  func.call @check_shrui(%cst0, %cst1) : (i16, i16) -> ()
+  func.call @check_shrui(%cst1, %cst1) : (i16, i16) -> ()
+  func.call @check_shrui(%cst1, %cst0) : (i16, i16) -> ()
+  func.call @check_shrui(%cst_1, %cst1) : (i16, i16) -> ()
+  func.call @check_shrui(%cst_1, %cst15) : (i16, i16) -> ()
+
+  // CHECK-NEXT: 1337
+  // CHECK-NEXT: 334
+  // CHECK-NEXT: 10
+  // CHECK-NEXT: 5
+  // CHECK-NEXT: 2
+  // CHECK-NEXT: 0
+  func.call @check_shrui(%cst1337, %cst0) : (i16, i16) -> ()
+  func.call @check_shrui(%cst1337, %cst2) : (i16, i16) -> ()
+  func.call @check_shrui(%cst1337, %cst7) : (i16, i16) -> ()
+  func.call @check_shrui(%cst1337, %cst8) : (i16, i16) -> ()
+  func.call @check_shrui(%cst1337, %cst9) : (i16, i16) -> ()
+  func.call @check_shrui(%cst1337, %cst15) : (i16, i16) -> ()
+
+  // CHECK-NEXT: 16384
+  // CHECK-NEXT: 8192
+  // CHECK-NEXT: 256
+  // CHECK-NEXT: 128
+  // CHECK-NEXT: 64
+  // CHECK-NEXT: 1
+  func.call @check_shrui(%cst_i16_min, %cst1) : (i16, i16) -> ()
+  func.call @check_shrui(%cst_i16_min, %cst2) : (i16, i16) -> ()
+  func.call @check_shrui(%cst_i16_min, %cst7) : (i16, i16) -> ()
+  func.call @check_shrui(%cst_i16_min, %cst8) : (i16, i16) -> ()
+  func.call @check_shrui(%cst_i16_min, %cst9) : (i16, i16) -> ()
+  func.call @check_shrui(%cst_i16_min, %cst15) : (i16, i16) -> ()
+
+  return
+}