mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
4b18ab869f
Summary: In non-strict mode of torch.export() we didn't set those `is_compiling()` to `True` which is needed by some models. Test Plan: Unit tests and manual testing. Differential Revision: D53624452 Pull Request resolved: https://github.com/pytorch/pytorch/pull/119602 Approved by: https://github.com/suo
25 lines
471 B
ReStructuredText
25 lines
471 B
ReStructuredText
.. currentmodule:: torch.compiler
|
|
|
|
.. automodule:: torch.compiler
|
|
|
|
.. _torch.compiler_api:
|
|
|
|
torch.compiler API reference
|
|
============================
|
|
|
|
For a quick overview of ``torch.compiler``, see :ref:`torch.compiler_overview`.
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
compile
|
|
reset
|
|
allow_in_graph
|
|
assume_constant_result
|
|
list_backends
|
|
disable
|
|
cudagraph_mark_step_begin
|
|
is_compiling
|
|
is_dynamo_compiling
|