add poly_to_llvm test file

j2kun · Oct 30, 2023 · f2c74c6 · f2c74c6
1 parent 3dc2169
commit f2c74c6
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/tests/poly_to_llvm.mlir b/tests/poly_to_llvm.mlir
@@ -0,0 +1,13 @@
+// RUN: tutorial-opt --poly-to-llvm %s | FileCheck %s
+
+// CHECK-NOT: poly
+func.func @test_poly_fn(%arg : i32) -> i32 {
+  %tens = tensor.splat %arg : tensor<10xi32>
+  %input = poly.from_tensor %tens : tensor<10xi32> -> !poly.poly<10>
+  %0 = poly.constant dense<[2, 3, 4]> : tensor<3xi32> : !poly.poly<10>
+  %1 = poly.add %0, %arg : !poly.poly<10>
+  %2 = poly.mul %1, %1 : !poly.poly<10>
+  %3 = poly.sub %2, %arg : !poly.poly<10>
+  %4 = poly.eval %3, %point: (!poly.poly<10>, i32) -> i32
+  return %4 : i32
+}