Skip to content

Commit a12dfad

Browse files
committed
lint fix
1 parent 056c6a3 commit a12dfad

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tilelang/language/proxy.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,10 @@ def _construct_strides(shape: Tuple[Any]):
150150
strides.append(s)
151151
return tuple(reversed(strides))
152152

153-
def __call__(self, shape: Union[Tuple[Any], PrimExpr, int], dtype: str = "float32", data=None) -> tir.Buffer:
153+
def __call__(self,
154+
shape: Union[Tuple[Any], PrimExpr, int],
155+
dtype: str = "float32",
156+
data=None) -> tir.Buffer:
154157
if isinstance(shape, (int, PrimExpr)):
155158
shape = (shape,)
156159
return super().__call__(

0 commit comments

Comments
 (0)