Skip to content

GitLab

Unverified Commit 30d8dedd authored by LJC00118's avatar LJC00118 Committed by GitHub
Browse files

[Bugfix] Minor fix in `builder.py` (#1235)

parent 8fbe1b3a
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tilelang/language/v2/builder.py
View file @ 30d8dedd
...@@ -390,6 +390,7 @@ class Builder(BaseBuilder): ...@@ -390,6 +390,7 @@ class Builder(BaseBuilder):
self.check_continue_break() self.check_continue_break()
if is_var(target): if is_var(target):
tir.buffer_store(target, eval_op(op, target[0], aug_value), 0) tir.buffer_store(target, eval_op(op, target[0], aug_value), 0)
return target
elif isinstance(target, Buffer): elif isinstance(target, Buffer):
raise RuntimeError("Augmented assignment is not supported for Buffer") raise RuntimeError("Augmented assignment is not supported for Buffer")
else: else:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment