Skip to content

Commit c57e26d

Browse files
committed
update parse error msg
1 parent 81a7354 commit c57e26d

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

python/tvm/script/parser.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -630,15 +630,18 @@ def transform_Assign(self, node):
630630
return tvm.tir.LetStmt(var, value, body, span=tvm_span_from_synr(node.span))
631631

632632
self.report_error(
633-
"""Assignments should be either
633+
"""Assignments should be one of:
634634
1. A "special statement" with return value
635635
1.1 Buffer = T.match_buffer()/T.buffer_decl()
636636
1.2 Var = T.var()
637637
1.3 Var = T.env_thread()
638638
2. A store into a buffer: Buffer[PrimExpr, PrimExpr, ..., PrimExpr] = PrimExpr
639639
3. A store into a variable: Var[PrimExpr] = PrimExpr
640640
4. A with scope handler with concise scoping and var def
641-
4.1 var = T.allocate()""",
641+
4.1 var = T.allocate()
642+
5. The right-hand side being a call to a pure python function, consuming and
643+
producing TVMScript values.
644+
x, y = f(...)""",
642645
node.span,
643646
)
644647

0 commit comments

Comments
 (0)