Update

fffrog · fffrog · commit 0f3cfbd97aed · 2024-11-19T22:43:24.000+08:00
[ghstack-poisoned]
diff --git a/torch/_C/__init__.pyi.in b/torch/_C/__init__.pyi.in
@@ -1355,7 +1355,6 @@ def __is_forward_AD_enabled() -> _bool: ...
 def _register_default_hooks(pack_hook: Callable, unpack_hook: Callable) -> None: ...
 def _reset_default_hooks() -> None: ...
 def _is_torch_function_mode_enabled() -> _bool: ...
-def _set_torch_function_mode(cls: Any) -> None: ...
 def _push_on_torch_function_stack(cls: Any) -> None: ...
 def _pop_torch_function_stack() -> Any: ...
 def _get_function_stack_at(idx: _int) -> Any: ...
