From d81b305bc681ca3cd7ff8930900731fcacc34f0b Mon Sep 17 00:00:00 2001 From: Chen-Pang He Date: Tue, 26 Nov 2024 20:38:50 +0000 Subject: [PATCH] Restore general handler for `aten.hardtanh` --- torch_ttnn/passes/lowering/to_tt_pass.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch_ttnn/passes/lowering/to_tt_pass.py b/torch_ttnn/passes/lowering/to_tt_pass.py index 0a05f134f..62f8948de 100644 --- a/torch_ttnn/passes/lowering/to_tt_pass.py +++ b/torch_ttnn/passes/lowering/to_tt_pass.py @@ -221,7 +221,7 @@ def __init__(self, target, args, kwargs): ############################################################ # Pointwise unary ############################################################ - if target == torch.ops.aten.hardtanh.default and args[1] == -1.0 and args[2] == 1.0: + if target == torch.ops.aten.hardtanh.default: # aten.hardtanh args are positional but ttnn.clip uses kw args new_kwargs = map_args_to_kwargs(args, ((1, "min_val"), (2, "max_val")), default_none=True) new_args = (args[0],)