Skip to content

Commit cd6e059

Browse files
committed
pre commit format
1 parent 3acce65 commit cd6e059

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

torch_ttnn/passes/lowering/to_tt_pass.py

+2-1
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,
@@ -1152,7 +1153,7 @@ def reshape_1d(code, args=args, kwargs=kwargs):
11521153
# slice_scatter could be concat([pre_slice_tensor, src_tensor, post_slice_tensor])
11531154
rank = len(tensor_shape)
11541155
[step] = step or [1]
1155-
1156+
11561157
assert dim < rank, f"The slice dim {dim} should be less than rank {rank}"
11571158

11581159
dim = (dim + rank) % rank

0 commit comments

Comments
 (0)