Skip to content

Commit

Permalink
Update guard for speecht5-tts model
Browse files Browse the repository at this point in the history
After folding torch.ops.aten.arange.start op, the guard rules
should be updated
  • Loading branch information
Po-Sheng Chang committed Nov 7, 2024
1 parent e297bef commit 1835c27
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions torch_ttnn/passes/lowering/to_tt_guard.py
Original file line number Diff line number Diff line change
Expand Up @@ -480,7 +480,7 @@
# add output is bf16 type, which is wrong

aten_slice_Tensor_blocklist += [
["Tensor<[24]> self = ?", "int dim = 0", "Optional[int] start = 0", "Optional[int] end = 9223372036854775807"],
["Tensor self = ?", "int dim = 0", "Optional[int] start = 0", "Optional[int] end = 9223372036854775807"],
[
"Tensor<[1, 24]> self = ?",
"int dim = 1",
Expand All @@ -490,7 +490,7 @@
]


aten_unsqueeze_default_blocklist += [["Tensor<[24]> self = ?", "int dim = 1"], ["Tensor<[24]> self = ?", "int dim = 0"]]
aten_unsqueeze_default_blocklist += [["Tensor<[24]> self = ?", "int dim = 1"], ["Tensor self = ?", "int dim = 0"]]

# ttnn.add
# shape=Shape([1, 24[32], 768]
Expand Down

0 comments on commit 1835c27

Please sign in to comment.