mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
500ce29e4c
with 20K features saves 20 seconds. 257.021589517593-> 237.8304626941681 buck2 run @fbcode//mode/opt fbcode//torchrec/distributed/tests:pt2_compile_benchmark -- --num-features=2000 Pull Request resolved: https://github.com/pytorch/pytorch/pull/140027 Approved by: https://github.com/ezyang
70 lines
1.5 KiB
ReStructuredText
70 lines
1.5 KiB
ReStructuredText
.. currentmodule:: torch.fx.experimental
|
|
|
|
torch.fx.experimental
|
|
=====================
|
|
|
|
.. warning::
|
|
These APIs are experimental and subject to change without notice.
|
|
|
|
torch.fx.experimental.symbolic_shapes
|
|
-------------------------------------
|
|
.. currentmodule:: torch.fx.experimental.symbolic_shapes
|
|
.. automodule:: torch.fx.experimental.symbolic_shapes
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
ShapeEnv
|
|
DimDynamic
|
|
StrictMinMaxConstraint
|
|
RelaxedUnspecConstraint
|
|
EqualityConstraint
|
|
SymbolicContext
|
|
StatelessSymbolicContext
|
|
StatefulSymbolicContext
|
|
SubclassSymbolicContext
|
|
DimConstraints
|
|
ShapeEnvSettings
|
|
ConvertIntKey
|
|
CallMethodKey
|
|
PropagateUnbackedSymInts
|
|
DivideByKey
|
|
InnerTensorKey
|
|
|
|
hint_int
|
|
is_concrete_int
|
|
is_concrete_bool
|
|
has_free_symbols
|
|
has_free_unbacked_symbols
|
|
definitely_true
|
|
definitely_false
|
|
guard_size_oblivious
|
|
sym_eq
|
|
constrain_range
|
|
constrain_unify
|
|
canonicalize_bool_expr
|
|
statically_known_true
|
|
lru_cache
|
|
check_consistent
|
|
compute_unbacked_bindings
|
|
rebind_unbacked
|
|
resolve_unbacked_bindings
|
|
is_accessor_node
|
|
|
|
torch.fx.experimental.proxy_tensor
|
|
-------------------------------------
|
|
|
|
.. currentmodule:: torch.fx.experimental.proxy_tensor
|
|
.. automodule:: torch.fx.experimental.proxy_tensor
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
make_fx
|
|
handle_sym_dispatch
|
|
get_proxy_mode
|
|
maybe_enable_thunkify
|
|
maybe_disable_thunkify
|