| 804 | /// \param arg function argument |
| 805 | /// \return function value stored in single-preicision |
| 806 | MEGDNN_HOST MEGDNN_DEVICE static expr exp(float arg) { |
| 807 | #if defined(__CUDA_ARCH__) |
| 808 | return expr(expf(arg)); |
| 809 | #else |
| 810 | return expr(std::exp(arg)); |
| 811 | #endif |
| 812 | } |
| 813 | |
| 814 | /// Exponential implementation. |
| 815 | /// \param arg function argument |