search

csarofeen / pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration
http://pytorch.org
Other
26 stars 7 forks source link

Add `aten::_softmax` parsing #1309

Closed kevinstephano closed 2 years ago

kevinstephano commented 2 years ago

🚀 Feature

Need to add parser support in parser.cpp for aten::_softmax op. LTC traces the _softmax op variant that is different from TorchScript.

kevinstephano commented 2 years ago

PR #1288 merged the code changes into NVFuser dev. Still waiting on PR to be upstreamed.