IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /max/get-started.md). For the complete documentation index, see llms.txt.
Skip to main content
For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).

Python class

CompileWrapper

CompileWrapper​

class max.pipelines.diffusion.CompileWrapper(compile_target, input_types=None)

source

Bases: object

Wraps a compile target with optional input type annotations.

Initialize the CompileWrapper.

Parameters:

  • compile_target (_CompileTarget) – The function or module to be compiled.
  • input_types (Iterable[TensorType] | None) – A list of input types (TensorTypes) required for compilation.

Raises:

ValueError – If input_types is not provided.