jafioti / luminal

Deep learning at the speed of light.

Home Page:https://luminalai.com

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ONNX Integration

jafioti opened this issue · comments

ONNX graphs should be able to be converted to luminal primgraphs. Only tricky part is getting GraphTensors out once the graph is converted, for inputting data and getting output.

This can probably just be implemented in a seperate crate luminal_onnx with a single function onnx_graph(file_name) -> Graph or something. Once the primgraph has been created from the onnx file, normal luminal compilers can be ran on it.

Currently I'm bandwidth constrained so if someone wants to pick this up, it's fairly isolated from the rest of the codebase so should be a decent first issue.