add poly.to_tensor op

j2kun · Oct 19, 2023 · aa076f5 · aa076f5
1 parent 8eea0d1
commit aa076f5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/lib/Dialect/Poly/PolyOps.td b/lib/Dialect/Poly/PolyOps.td
@@ -45,6 +45,13 @@ def Poly_FromTensorOp : Op<Poly_Dialect, "from_tensor", [Pure]> {
   let hasFolder = 1;
 }
 
+def Poly_ToTensorOp : Op<Poly_Dialect, "to_tensor", [Pure]> {
+  let summary = "Converts a polynomial to a tensor of its integer coefficients.";
+  let arguments = (ins Polynomial:$input);
+  let results = (outs TensorOf<[AnyInteger]>:$output);
+  let assemblyFormat = "$input attr-dict `:` qualified(type($input)) `->` type($output)";
+}
+
 def IntOrComplex : AnyTypeOf<[AnyInteger, AnyComplex]>;
 
 def Poly_EvalOp : Op<Poly_Dialect, "eval", [AllTypesMatch<["point", "output"]>, Has32BitArguments]> {

diff --git a/tests/poly_syntax.mlir b/tests/poly_syntax.mlir
@@ -39,6 +39,9 @@ module {
     %12 = poly.constant dense<"0x020304"> : tensor<3xi8> : !poly.poly<10>
     %13 = poly.constant dense<4> : tensor<100xi32> : !poly.poly<10>
 
+    // CHECK: poly.to_tensor
+    %14 = poly.to_tensor %1 : !poly.poly<10> -> tensor<10xi32>
+
     return %4 : !poly.poly<10>
   }
 }