spcl / daceml

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

Decorator option to compile only child modules #84

Closed tbennun closed 2 years ago

tbennun commented 3 years ago

in order to save parameters for individual modules and avoid issues with PyTorch and 64 allowed parameters per module call.

Alternatively: mode that runs with ctypes/pybind11 when too many parameters are present.