MCPcopy Create free account
hub / github.com/AI-Hypercomputer/maxtext / aot_compile

Method aot_compile

src/MaxText/prefill_packing.py:105–108  ·  view source on GitHub ↗

Ahead-of-time compile prefill processing routines.

(self, params: Params, input_padding: int)

Source from the content-addressed store, hash-verified

103 self.process_func = {}
104
105 def aot_compile(self, params: Params, input_padding: int):
106 """Ahead-of-time compile prefill processing routines."""
107
108 return self._process_compiled(params, input_padding)
109
110 def process(
111 self,

Callers 1

run_benchmarksFunction · 0.95

Calls 1

_process_compiledMethod · 0.95

Tested by

no test coverage detected