[Mlir-commits] [mlir] [mlir][NVVM] Tighten result-type predicate on special-register ops (PR #195030)

[Guray Ozen]

================
@@ -306,22 +306,28 @@ class NVVM_SingleResultIntrinsicOp<string mnemonic, list<Trait> traits = [], str
 // NVVM special register op definitions
 //===----------------------------------------------------------------------===//
 
-class NVVM_PureSpecialRegisterOp<string mnemonic, list<Trait> traits = []> :
+class NVVM_PureSpecialRegisterOp<string mnemonic, list<Trait> traits = [],
+                                 Type resultType = I32> :
   NVVM_IntrOp<mnemonic, !listconcat(traits, [Pure]), 1> {
   let arguments = (ins);
+  let results = (outs resultType:$res);
   let assemblyFormat = "attr-dict `:` type($res)";
 }
 
-class NVVM_SpecialRegisterOp<string mnemonic, list<Trait> traits = []> :
+class NVVM_SpecialRegisterOp<string mnemonic, list<Trait> traits = [],
+                             Type resultType = I32> :
----------------
grypp wrote:

nit: one can swap last two
```suggestion
class NVVM_SpecialRegisterOp<string mnemonic, Type resultType = I32, 
                                              list<Trait> traits = []> :
```

https://github.com/llvm/llvm-project/pull/195030