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

Function save_compiled

src/MaxText/train_compile.py:140–144  ·  view source on GitHub ↗

Serialize and save the compiled function.

(compiled, save_name)

Source from the content-addressed store, hash-verified

138
139
140def save_compiled(compiled, save_name):
141 """Serialize and save the compiled function."""
142 serialized, _, _ = serialize(compiled)
143 with open(save_name, "wb") as f:
144 pickle.dump(serialized, f)
145
146
147def is_oom(argv: Sequence[str]) -> bool:

Callers 1

mainFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected