MCPcopy Create free account
hub / github.com/apple/axlearn / lower

Method lower

axlearn/common/debug_utils.py:57–60  ·  view source on GitHub ↗

Traces and lowers the function using the provided arguments.

(self, *args, **kwargs)

Source from the content-addressed store, hash-verified

55 return compiled(*args, **kwargs)
56
57 def lower(self, *args, **kwargs) -> _CheckifyLoweredFnWrapper:
58 """Traces and lowers the function using the provided arguments."""
59 lowered = self._checkified_jit_handle.lower(*args, **kwargs)
60 return _CheckifyLoweredFnWrapper(lowered)
61
62
63class CheckifyJitFn(Protocol):

Callers 15

__call__Method · 0.95
from_flagsMethod · 0.80
_build_containerMethod · 0.80
_build_podMethod · 0.80
get_pathways_tpu_versionFunction · 0.80
test_delete_blobMethod · 0.80
test_upload_blobMethod · 0.80
_node_pool_bodyFunction · 0.80
_get_statusMethod · 0.80
_executeMethod · 0.80

Calls 1