search

microsoft / onnxscript

ONNX Script enables developers to naturally author ONNX functions and models using a subset of Python.
https://onnxscript.ai/
MIT License
246 stars 46 forks source link

IR Integration plan #1384

Open justinchuby opened 2 months ago

justinchuby commented 2 months ago 
flowchart TD
    pattern(Rewriter pattern)  --> generic_pattern(Unify pattern classes) --> passes(Rewriter passes)
    script(Replace ONNX Script IR) --> value(Change `Op.__cal__` to return `Value`)
    script(Replace ONNX Script IR) --> dtype(Update and unify the data types)
    torchlib(Update torchlib to use IR) --> trace(Unify with ONNX Script trace mode)
justinchuby commented 2 months ago