You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
1.4 KiB
46 lines
1.4 KiB
"""
|
|
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__])
|