| 123456789101112131415161718192021222324252627282930313233343536373839404142434445 |
- """
- Configuration module for torch.export.export.
- This module contains various configuration flags and settings that control torch.export's
- behavior, including:
- - Runtime behavior flags
- - Debugging and development options
- """
- import sys
- from typing import Any, TYPE_CHECKING
- from torch._environment import is_fbcode
- from torch.utils._config_module import install_config_module
- # this flag controls whether we use new functional tracer. It
- # should be True in the long term.
- use_new_tracer_experimental = True
- # this flag is used to control whether we want to instrument
- # fake tensor creation to track potential leaks. It is off
- # by default, but user can turn it on to debug leaks.
- detect_non_strict_fake_tensor_leaks = False
- # error on potentially pre-dispatch/non-strict tracing limitation
- # this type of error usually happens when we encounter an op
- # that we don't know how to proxy, resulting in untracked fake tensors
- error_on_lifted_constant_tensors = True
- # enable auto_functionalized_v2 in export
- # We turn this off in fbcode due to downstream users not
- # being ready to handle auto_functionalized_v2.
- enable_auto_functionalized_v2_for_export = not is_fbcode()
- use_legacy_dynamo_graph_capture = True
- if TYPE_CHECKING:
- from torch.utils._config_typing import * # noqa: F401, F403
- def _make_closure_patcher(**changes: Any) -> Any: ...
- install_config_module(sys.modules[__name__])
|
