Dynamo

  • ThunderCompiler — A class that compiles a torch.fx.GraphModule to a thunder.ThunderModule. This class is meant to be used as a backend for the torch.compile() function.
  • thunderfx — Compiles a callable (function or model) by using Thunder as the backend of torch.compile().
  • ThunderFXCompiledObject — A wrapper around the result of thunderfx() compilation.

Was this page helpful?

What went wrong?

↑↓ Navigate↵ Open⌘↵ New tabesc Close