From 1835c2702ae6c05176864be06f1b5b2224b2b7de Mon Sep 17 00:00:00 2001 From: Po-Sheng Chang Date: Thu, 7 Nov 2024 08:04:32 +0000 Subject: [PATCH] Update guard for speecht5-tts model After folding torch.ops.aten.arange.start op, the guard rules should be updated --- torch_ttnn/passes/lowering/to_tt_guard.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torch_ttnn/passes/lowering/to_tt_guard.py b/torch_ttnn/passes/lowering/to_tt_guard.py index 905e0f1c..817322dc 100644 --- a/torch_ttnn/passes/lowering/to_tt_guard.py +++ b/torch_ttnn/passes/lowering/to_tt_guard.py @@ -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", @@ -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]