Skip to content

Commit

Permalink
Fix the link to torch.compiler_custom_backends. (#125865)
Browse files Browse the repository at this point in the history
Tiny fix. Fixes #119272

Pull Request resolved: #125865
Approved by: https://github.com/soulitzer
  • Loading branch information
zeb209 authored and pytorchmergebot committed May 11, 2024
1 parent 0a9c6e9 commit c1690a3
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion torch/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -1834,7 +1834,8 @@ def compile(model: Optional[Callable] = None, *,
- Experimental or debug in-tree backends can be seen with `torch._dynamo.list_backends(None)`
- To register an out-of-tree custom backend: https://pytorch.org/docs/main/torch.compiler_custom_backends.html
- To register an out-of-tree custom backend:
https://pytorch.org/docs/main/torch.compiler_custom_backends.html#registering-custom-backends
mode (str): Can be either "default", "reduce-overhead", "max-autotune" or "max-autotune-no-cudagraphs"
- "default" is the default mode, which is a good balance between performance and overhead
Expand Down

0 comments on commit c1690a3

Please sign in to comment.