-
Notifications
You must be signed in to change notification settings - Fork 180
Open
Description
We need to have support for importing StableHLO graphs into Luminal. This will bring PyTorch and JAX support, and be a good pattern to use for ONNX #32
What I think this looks like is a crate called luminal_hlo which has one function, import_hlo(file: String) -> (Graph, Vec<GraphTensor>). The input might be a different format but the general idea is we have a single function that yields a luminal graph and a set of graph tensors hooking into it.
The output graph will be a primitive graph, and then compilers can be ran on it to optimize.
Metadata
Metadata
Assignees
Labels
No labels