| 1191 | /// \param arg function argument |
| 1192 | /// \return function value stored in single-preicision |
| 1193 | MEGDNN_HOST MEGDNN_DEVICE static expr exp(float arg) { |
| 1194 | #if defined(__CUDA_ARCH__) |
| 1195 | return expr(expf(arg)); |
| 1196 | #else |
| 1197 | return expr(std::exp(arg)); |
| 1198 | #endif |
| 1199 | } |
| 1200 | |
| 1201 | /// Exponential implementation. |
| 1202 | /// \param arg function argument |