use struct for type

Author: j2kun

mlir/test/Dialect/Polynomial/canonicalization.mlir

@@ -1,7 +1,7 @@
 // RUN: mlir-opt -canonicalize %s | FileCheck %s
 #ntt_poly = #polynomial.int_polynomial<-1 + x**8>
 #ntt_ring = #polynomial.ring<coefficientType=i32, coefficientModulus=256, polynomialModulus=#ntt_poly, primitiveRoot=31>
-!ntt_poly_ty = !polynomial.polynomial<#ntt_ring>
+!ntt_poly_ty = !polynomial.polynomial<ring=#ntt_ring>
 !tensor_ty = tensor<8xi32, #ntt_ring>
 
 // CHECK-LABEL: @test_canonicalize_intt_after_ntt
@@ -34,16 +34,17 @@ func.func @test_canonicalize_ntt_after_intt(%t0 : !tensor_ty) -> !tensor_ty {
 #ring = #polynomial.ring<coefficientType=i32, coefficientModulus=256:i32, polynomialModulus=#cycl_2048>
 #one_plus_x_squared = #polynomial.int_polynomial<1 + x**2>
 #one_minus_x_squared = #polynomial.int_polynomial<1 + -1x**2>
+!sub_ty = !polynomial.polynomial<ring=#ring>
 
 // CHECK-LABEL: test_canonicalize_sub_power_of_two_cmod
-func.func @test_canonicalize_sub_power_of_two_cmod() -> !polynomial.polynomial<#ring> {
-  %poly0 = polynomial.constant {value=#one_plus_x_squared} : !polynomial.polynomial<#ring>
-  %poly1 = polynomial.constant {value=#one_minus_x_squared} : !polynomial.polynomial<#ring>
-  %0 = polynomial.sub %poly0, %poly1  : !polynomial.polynomial<#ring>
+func.func @test_canonicalize_sub_power_of_two_cmod() -> !sub_ty {
+  %poly0 = polynomial.constant {value=#one_plus_x_squared} : !sub_ty
+  %poly1 = polynomial.constant {value=#one_minus_x_squared} : !sub_ty
+  %0 = polynomial.sub %poly0, %poly1  : !sub_ty
   // CHECK: %[[minus_one:.+]] = arith.constant -1 : i32
   // CHECK: %[[p1:.+]] = polynomial.constant
   // CHECK: %[[p2:.+]] = polynomial.constant
   // CHECK: %[[p2neg:.+]] = polynomial.mul_scalar %[[p2]], %[[minus_one]]
   // CHECK: [[ADD:%.+]] = polynomial.add %[[p1]], %[[p2neg]]
-  return %0 : !polynomial.polynomial<#ring>
+  return %0 : !sub_ty
 }