[mlir] Add InferTensorType without supporting reifyReturnTypeShapes

[Chia-hungDuan]

Summary

This is useful for the case that we don't need to implement
reifyReturnTypeShapes.

[jpienaar]

SG, one could also just do InferTensorType<> previously I believe where this would allow InferTensorType without <>.

I actually wonder if the above needs the extra .traits, may have been a point in time, I'd expect

def InferTensorTypeWithReify : InferTensorTypeBase<"reifyReturnTypeShapes">;

to be sufficient.

[Chia-hungDuan]

Rebase and address review comment

[Chia-hungDuan]

In D121403#3380441, @jpienaar wrote:

SG, one could also just do InferTensorType<> previously I believe where this would allow InferTensorType without <>.

I actually wonder if the above needs the extra .traits, may have been a point in time, I'd expect

def InferTensorTypeWithReify : InferTensorTypeBase<"reifyReturnTypeShapes">;

to be sufficient.

Yes, that works and it's cleaner.

mlir/include/mlir/Interfaces/InferTypeOpInterface.td

InterfaceMethod<
       /*methodBody=*/[{}],
       /*defaultImplementation=*/[{ return ::mlir::failure(); }]
     >
   ];
 }
 
 // Convenience class grouping together type and shaped type op interfaces for
 // ops that have tensor return types.
-class InferTensorType<list<string> overridenMethods = []> : TraitList<
+class InferTensorTypeBase<list<string> overridenMethods = []> : TraitList<
   [
     // Op implements infer type op interface.
     InferTypeOpInterface,
     // The op will have methods implementing the ShapedType type inference
     // interface.
     DeclareOpInterfaceMethods<InferShapedTypeOpInterface, overridenMethods>,
     // The op produces tensors and will use the ShapedType type infer interface
     // along with knowledge that it is producing Tensors to infer the type.
     NativeOpTrait<"InferTensorType">
   ]>;
-def InferTensorTypeWithReify: TraitList<InferTensorType<[
-    "inferReturnTypeComponents", "reifyReturnTypeShapes"]>.traits>;
 
+def InferTensorType : InferTensorTypeBase<["inferReturnTypeComponents"]>;
+def InferTensorTypeWithReify: InferTensorTypeBase<[
+    "inferReturnTypeComponents", "reifyReturnTypeShapes"]>;
 
 def ReifyRankedShapedTypeOpInterface :
     OpInterface<"ReifyRankedShapedTypeOpInterface"> {
   let description = [{
     Interface to compute the shape of the result of an operation when
     the result is a ranked shape type, i.e. `RankedTensorType` or
     `MemRefType`.
   }];