Method compile

axlearn/common/debug_utils.py:32–37

Compile the function with provided options.

(
        self, compiler_options: Optional[dict[str, Union[str, bool]]] = None
    )

        self._lowered = lowered

    def compile(
        self, compiler_options: Optional[dict[str, Union[str, bool]]] = None
    ) -> _CheckifyCompiledFnWrapper:
        """Compile the function with provided options."""
        compiled = self._lowered.compile(compiler_options=compiler_options)
        return _CheckifyCompiledFnWrapper(compiled)


class CheckifyJitFnWrapper: