Restore general handler for aten.hardtanh

tenstorrent · Dec 10, 2024 · d81b305 · d81b305
1 parent 4e1cbfe
commit d81b305
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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],)