Fix bug in SoftmaxOp, compatible with MXNet 2.0, add custom Op C++ helper, refactor omp context - Githubissues

wkcn / MobulaOP

A Simple & Flexible Cross Framework Operators Toolkit

MIT License

164 stars 21 forks source link

Fix bug in SoftmaxOp, compatible with MXNet 2.0, add custom Op C++ helper, refactor omp context #97

Closed wkcn closed 4 years ago

wkcn commented 4 years ago

In MXNet 2.0, mx.sym.simple_bind has been renamed to mx.sym._simple_bind.

[x] compatible with MXNet 2.0
[x] add CustomOp C++ helper
[x] Fix bug in Softmax Op
[x] refactor omp context