Showing content from https://github.com/jax-ml/jax/issues/7733 below:
ahead-of-time lowering and compilation · Issue #7733 · jax-ml/jax · GitHub
Skip to content Navigation Menu
Saved searches Use saved searches to filter your results more quickly
Sign up You signed in with another tab or window. Reload to refresh your session. You signed out in another tab or window. Reload to refresh your session. You switched accounts on another tab or window. Reload to refresh your session. Dismiss alert Additional navigation options
ahead-of-time lowering and compilation #7733
Description
Following the experiment in #6034, plus other thoughts and motivation accrued since, we should support a notion of lowering and compilation "ahead of time," i.e. upon explicit request. Some things to note:
- Lowering means producing XLA HLO. Compiling means running it through the XLA compiler
- Compiling produces optimized HLO and renders a computation ready for execution.
- Lowering and compilation happen for a fixed type signature.
- A function compiled ahead of time and can be called with arguments of the fixed type signature only.
- A function compiled ahead of time will intentionally fail to transform, e.g. if called under
jvp or vmap, since it will be compiled for a particular type signature, and transformations modify signatures.
- We will want this for every compiling transformation:
jit, pjit, pmap, xmap.
- Although a function compiled ahead of time is bound to correspond to some XLA executable, it's out of scope to provide supported access to, or any serialization of, that executable.
You can’t perform that action at this time.
RetroSearch is an open source project built by @garambo
| Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.3