[Mlir-commits] [mlir] [mlir][NVVM] Add InferTypeOpInterface to sync and ldmatrix ops (PR #188238)

[Pradeep Kumar]

================
@@ -2244,6 +2244,21 @@ LogicalResult ShflOp::verify() {
   return success();
 }
 
+LogicalResult ShflOp::inferReturnTypes(
+    MLIRContext *context, std::optional<Location> location,
+    ShflOp::Adaptor adaptor,
+    SmallVectorImpl<Type> &inferredReturnTypes) {
+  Type valType = adaptor.getVal().getType();
+  if (adaptor.getReturnValueAndIsValid()) {
+    SmallVector<Type> body = {valType, IntegerType::get(context, 1)};
----------------
schwarzschild-radius wrote:

Can we replace the vector with the init expression?

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