Skip to content

GitLab

"src/vscode:/vscode.git/clone" did not exist on "ba0de4b129ffe6469d95291f36c2c50b6fd0fb92"
Unverified Commit b0206854 authored by Kuris's avatar Kuris Committed by GitHub
Browse files

[Fix] fix wrong uint narrowing bug in tvm in #1310 (#1320)

parent 9dda774a
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
tvm @ 3354ada7
Compare cd2b2b60...3354ada7
Subproject commit cd2b2b6013d155b5822300b0a0740fa65320dd9e Subproject commit 3354ada79dd428e383102020814fa9c37638e752
tilelang/language/allocate.py
View file @ b0206854
...@@ -22,6 +22,7 @@ from tvm.tir import PrimExpr ...@@ -22,6 +22,7 @@ from tvm.tir import PrimExpr
from tvm.script.parser.tir import block_attr from tvm.script.parser.tir import block_attr
from tvm.tir.buffer import Buffer from tvm.tir.buffer import Buffer
from tvm.tir.expr import FloatImm, IntImm from tvm.tir.expr import FloatImm, IntImm
from .v2.dtypes import dtype as tl_dtype
def alloc_shared(shape, dtype, scope="shared.dyn"): def alloc_shared(shape, dtype, scope="shared.dyn"):
...@@ -135,7 +136,7 @@ def alloc_var(dtype, *args, scope="local.var", init: PrimExpr | None = None): ...@@ -135,7 +136,7 @@ def alloc_var(dtype, *args, scope="local.var", init: PrimExpr | None = None):
buffer = T.alloc_buffer([1], dtype, scope=parsed_scope) buffer = T.alloc_buffer([1], dtype, scope=parsed_scope)
if parsed_init is not None: if parsed_init is not None:
if isinstance(parsed_init, (int, float, IntImm, FloatImm)): if isinstance(parsed_init, (int, float, IntImm, FloatImm)):
block_attr({"tl.local_var_init": {buffer.data: parsed_init}}) block_attr({"tl.local_var_init": {buffer.data: tl_dtype(dtype)(parsed_init)}})
else: else:
T.buffer_store(buffer, parsed_init, 0) T.buffer_store(buffer, parsed_init, 0)
return buffer return buffer
......
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