Conversion

namespace mamba_ssm

namespace conversion

Functions

inline __device__ float toFloat (float f)

inline __device__ float toFloat (__half h)

inline __device__ float toFloat (__nv_bfloat16 val)

inline __device__ void convertAndStore (float *output, float input)

inline __device__ void convertAndStore (__half *output, float input)

inline __device__ void convertAndStore (__nv_bfloat16 *output, float input)