mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
4c8718d8e7
Attempt # 2 at https://github.com/pytorch/pytorch/pull/132926 to implement https://github.com/pytorch/pytorch/issues/123771. Implement a new `torch.compiler.set_stance` function that can force `torch.compile` regions to run eagerly. See added tests for usage examples. Pull Request resolved: https://github.com/pytorch/pytorch/pull/137504 Approved by: https://github.com/yf225, https://github.com/jansel
27 lines
512 B
ReStructuredText
27 lines
512 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
|
|
substitute_in_graph
|
|
assume_constant_result
|
|
list_backends
|
|
disable
|
|
set_stance
|
|
cudagraph_mark_step_begin
|
|
is_compiling
|
|
is_dynamo_compiling
|