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]