| 826 | /// \param arg function argument |
| 827 | /// \return function value stored in single-preicision |
| 828 | MEGDNN_HOST MEGDNN_DEVICE static expr exp2(float arg) { |
| 829 | #if defined(__CUDA_ARCH__) |
| 830 | return expr(exp2f(arg)); |
| 831 | #else |
| 832 | return expr(std::exp2(arg)); |
| 833 | #endif |
| 834 | } |
| 835 | |
| 836 | /// Logarithm implementation. |
| 837 | /// \param arg function argument |