A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.
952
stars
158
forks
source link
Support JIT for class and class method + nnf config + 3-lvl-signature #398
Closed
siahuat0727 closed 2 years ago
As discussed,
Added
Support JIT for class method (as decorator and as function)
model = Net() model.forward = nnfusion.jit(model.forward)
forward
will have the same behavior as
nnfusion.Config
to fully leverage nnfusion-cliChanged
torch.nn.Module
(instead of returning a function).__module__
I'd love to hear your feedback.
392