Skip to content

Commit e295ba1

Browse files
committed
lint fix
1 parent 7880876 commit e295ba1

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tilelang/language/reduce.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,12 +24,11 @@ def reduce(buffer: tir.Buffer, out: tir.Buffer, reduce_type: str, dim: int, clea
2424
Returns:
2525
tir.Call: Handle to the reduction operation
2626
"""
27-
expected_shape = buffer.shape[:dim] + buffer.shape[dim+1:]
27+
expected_shape = buffer.shape[:dim] + buffer.shape[dim + 1:]
2828
if expected_shape != list(out.shape):
2929
raise ValueError(
3030
f"Invalid reduce output shape, buffer shape is {buffer.shape}, dim is {dim}, "
31-
f"expected output shape {expected_shape}, got output shape {out.shape}"
32-
)
31+
f"expected output shape {expected_shape}, got output shape {out.shape}")
3332
buffer = buffer.access_ptr("r")
3433
out = out.access_ptr("w")
3534
return tir.call_intrin(

0 commit comments

Comments
 (0)