mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
a8a44a1aa2
This PR: - adds deprecation warnings when calling the functorch APIs - adds documentation saying that those APIs are deprecated It does this by creating thin wrappers around the original APIs that (1) raise deprecation warnings and (2) have an additional line in their documentation that they are deprecated. NB: - Python surpresses DeprecationWarning, so we use UserWarning instead. Test Plan: - New tests - the functorch.* APIs are still tested for correctness because that's what test/functorch/* use (as opposed to directly calling the torch.func.* APIs) Pull Request resolved: https://github.com/pytorch/pytorch/pull/92279 Approved by: https://github.com/albanD, https://github.com/soulitzer
29 lines
1,013 B
Python
29 lines
1,013 B
Python
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the BSD-style license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
import torch
|
|
from . import _C
|
|
|
|
# Top-level APIs. Please think carefully before adding something to the
|
|
# top-level namespace:
|
|
# - private helper functions should go into torch._functorch
|
|
# - very experimental things should go into functorch.experimental
|
|
# - compilation related things should go into functorch.compile
|
|
|
|
# Was never documented
|
|
from torch._functorch.python_key import make_fx
|
|
|
|
from torch._functorch.deprecated import (
|
|
vmap, grad, grad_and_value, vjp, jvp, jacrev, jacfwd, hessian, functionalize,
|
|
make_functional, make_functional_with_buffers, combine_state_for_ensemble,
|
|
)
|
|
|
|
# utilities. Maybe these should go in their own namespace in the future?
|
|
from torch._functorch.make_functional import (
|
|
FunctionalModule,
|
|
FunctionalModuleWithBuffers,
|
|
)
|
|
|
|
__version__ = torch.__version__
|