Skip to content

Commit 9cbbad7

Browse files
committed
pre commit format
1 parent 576fa1f commit 9cbbad7

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

torch_ttnn/passes/lowering/to_tt_pass.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
import math
77
from torch._guards import detect_fake_mode
88
from torch._subclasses.fake_tensor import unset_fake_temporarily
9+
910
from torch_ttnn.utils import (
1011
GraphCleanup,
1112
TtnnBfloat16,
@@ -1146,7 +1147,7 @@ def reshape_1d(code, args=args, kwargs=kwargs):
11461147
# slice_scatter could be concat([pre_slice_tensor, src_tensor, post_slice_tensor])
11471148
rank = len(tensor_shape)
11481149
[step] = step or [1]
1149-
1150+
11501151
assert dim < rank, f"The slice dim {dim} should be less than rank {rank}"
11511152

11521153
dim = (dim + rank) % rank

0 commit comments

Comments
 (0)