mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
54976bca10
Collection of ready-made guard filters. One issue is that they are not composable - `filter1(filter2(guard))`. On the other hand, they are easy to use. Pull Request resolved: https://github.com/pytorch/pytorch/pull/155083 Approved by: https://github.com/zhxchen17, https://github.com/jansel
33 lines
684 B
Markdown
33 lines
684 B
Markdown
```{eval-rst}
|
|
.. 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`.
|
|
|
|
```{eval-rst}
|
|
.. 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
|
|
is_exporting
|
|
skip_guard_on_inbuilt_nn_modules_unsafe
|
|
skip_guard_on_all_nn_modules_unsafe
|
|
keep_tensor_guards_unsafe
|
|
skip_guard_on_globals_unsafe
|
|
```
|