Fix parser code

llvm · Feb 17, 2025 · 35a43fc · 35a43fc
1 parent 8275639
commit 35a43fc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/lib/Dialect/Torch/IR/TorchOps.cpp b/lib/Dialect/Torch/IR/TorchOps.cpp
@@ -5973,6 +5973,10 @@ ParseResult OnnxVariantRotaryEmbeddingOp::parse(OpAsmParser &parser,
     return failure();
   }
 
+  // Parse results type list.
+  if (parser.parseArrowTypeList(result.types))
+    return failure();
+
   return success();
 }