commit a15c02562a16928dd0bf9e2dfb10f1f6c7fdb1dd
author Tugsbayasgalan (Tugsuu) Manlaibaatar <tmanlaibaatar@meta.com> Tue Mar 05 03:19:59 2024 +0000
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Tue Mar 05 03:19:59 2024 +0000
tree 24d7fe8258a4dd84aee9f098cda16c3bed623c1d
parent 3381f282c35b61ca991eb84c0e938667244cbc7a

Fix dynamo failure (#121167)

Summary: Title

Test Plan: CI

Differential Revision: D54509198

Pull Request resolved: https://github.com/pytorch/pytorch/pull/121167
Approved by: https://github.com/izaitsevfb

torch/_dynamo/trace_rules.py

diff --git a/torch/_dynamo/trace_rules.py b/torch/_dynamo/trace_rules.py
index 49080d3..5c6ad0b 100644
--- a/torch/_dynamo/trace_rules.py
+++ b/torch/_dynamo/trace_rules.py
@@ -2213,7 +2213,6 @@
         "torch._higher_order_ops.out_dtype.out_dtype_fake_tensor_mode",
         "torch._higher_order_ops.out_dtype.out_dtype_fallback",
         "torch._higher_order_ops.out_dtype.out_dtype_func",
-        "torch._higher_order_ops.out_dtype.out_dtype_predispatch",
         "torch._higher_order_ops.out_dtype.out_dtype_proxy",
         "torch._higher_order_ops.out_dtype.trace_out_dtype",
         "torch._higher_order_ops.utils.autograd_not_implemented_inner",