CUDA/HIP: also specialize conversion nv_bfloat16 <-> int

Author: IMbackK

ggml/src/ggml-cuda/convert.cuh

@@ -70,3 +70,15 @@ __host__ __device__ inline nv_bfloat16 ggml_cuda_convert_val<half, nv_bfloat16>(
 {
     return __float2bfloat16(float(x));
 }
+
+template<>
+__host__ __device__ inline int ggml_cuda_convert_val<nv_bfloat16, int>(nv_bfloat16 x)
+{
+    return int(__bfloat162float(x));
+}
+
+template<>
+__host__ __device__ inline nv_bfloat16 ggml_cuda_convert_val<int, nv_bfloat16>(int x)
+{
+    return __float2bfloat16(float(x));
+}