Skip to content

StableHLO support #85

@jafioti

Description

@jafioti

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

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions