search

csarofeen / pytorch

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

Llu/softmax #2572

Closed liqiangxl closed 1 year ago

liqiangxl commented 1 year ago

Fixes #2125

csarofeen commented 1 year ago

what's the perf delta for the softmax case?

Sorry didn't meant to close the PR, reopened it.

liqiangxl commented 1 year ago

move to https://github.com/NVIDIA/Fuser/pull/11