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

Function _array_copy

src/MaxText/benchmark_chunked_prefill.py:60–61  ·  view source on GitHub ↗
(x)

Source from the content-addressed store, hash-verified

58@jax.jit
59def _copy(cache):
60 def _array_copy(x):
61 return x.copy()
62
63 return jax.tree.map(_array_copy, cache)
64

Callers

nothing calls this directly

Calls 1

copyMethod · 0.80

Tested by

no test coverage detected