spcl / daceml

A Data-Centric Compiler for Machine Learning
https://daceml.readthedocs.io
BSD 3-Clause "New" or "Revised" License
81 stars 15 forks source link

Clean up ONNX library node implementation #94

Open tbennun opened 2 years ago

tbennun commented 2 years ago

At the moment, creating ONNX library node implementations entails a double nested loop over all expansions, which is hard to read and debug. If we could avoid dynamic class creation and the double for loop, or at least clean it up, it would be more approachable.