Skip to content

Commit 0aad651

Browse files
author
nicunxiao
committed
new format
1 parent cda16d5 commit 0aad651

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tilelang/language/allocate.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
from __future__ import annotations
12
"""Memory allocation utilities for Tile-AI programs.
23
34
This module provides a set of functions for allocating different types of memory buffers
@@ -67,7 +68,7 @@ def alloc_fragment(shape, dtype, scope="local.fragment"):
6768
return T.alloc_buffer(shape, dtype, scope=scope)
6869

6970

70-
def alloc_var(dtype, *args, scope="local.var", init: Union[PrimExpr] = None):
71+
def alloc_var(dtype, *args, scope="local.var", init: Union[PrimExpr] = None): # noqa: UP007
7172
"""Allocate a single-element variable buffer.
7273
7374
Args:

0 commit comments

Comments
 (0)