Fix parser

Author: amd-eochoalo

mlir/lib/Dialect/AMDGPU/IR/AMDGPUDialect.cpp

@@ -52,16 +52,22 @@ struct AMDGPUInlinerInterface final : DialectInlinerInterface {
 
 static ParseResult
 parseDynamicIndex(OpAsmParser &parser,
-                  std::optional<OpAsmParser::UnresolvedOperand> dynamicSize,
+                  std::optional<OpAsmParser::UnresolvedOperand> &dynamicSize,
                   IntegerAttr &staticSize) {
 
-  int64_t staticVal = 0;
-  if (parser.parseOptionalInteger(staticVal).has_value()) {
+  int64_t staticVal;
+  OptionalParseResult parseResult = parser.parseOptionalInteger(staticVal);
+  if (parseResult.has_value()) {
     staticSize = parser.getBuilder().getIndexAttr(staticVal);
     return success();
   }
-  
-  return parser.parseOperand(dynamicSize.value());
+
+  OpAsmParser::UnresolvedOperand operand = OpAsmParser::UnresolvedOperand{};
+  if (parser.parseOperand(operand)) {
+    dynamicSize = operand;
+    return success();
+  }
+  return failure();
 }
 
 static void printDynamicIndex(OpAsmPrinter &printer, Operation *op,

mlir/test/Dialect/AMDGPU/ops.mlir

@@ -730,7 +730,7 @@ func.func @make_dma_descriptor(%base: !amdgpu.tdm_base<i32>, %barrier: memref<8x
         globalStride [1]
         // CHECK-SAME: sharedSize [0]
         sharedSize [0]
-        // CHECK-SAME: padShared(1 every 1)
+        // CHECK-SAME: padShared(%[[IDX]] every %[[IDX]])
         padShared(%idx every %idx)
         : !amdgpu.tdm_base<i32> -> !amdgpu.tdm_descriptor