thunderfx
1 parent 45a2e93 commit 4f64d85Copy full SHA for 4f64d85
2 files changed
docs/source/reference/dynamo/index.rst
@@ -7,3 +7,4 @@ thunder.dynamo
7
:toctree:
8
9
ThunderCompiler
10
+ thunderfx
thunder/dynamo/compiler.py
@@ -207,7 +207,8 @@ def save_reproducer_to_folder(
207
208
209
def thunderfx(fn: Callable, /, **kwargs) -> Callable:
210
- """Compiles a callable (function or model) by using Thunder as the backend of :func:`torch.compile`
+ """Compiles a callable (function or model) by using Thunder as the backend of :func:`torch.compile`.
211
+
212
Args:
213
fn: A :class:`~torch.nn.Module` or a function to compile.
214
Keyword Args:
0 commit comments