From 6486a6d7b29b6f89a8a23a696cf6614ac14f0b36 Mon Sep 17 00:00:00 2001 From: laurent Date: Thu, 29 Jun 2023 23:23:44 +0100 Subject: [PATCH] Avoid some cast kernels. --- candle-kernels/src/cast.cu | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/candle-kernels/src/cast.cu b/candle-kernels/src/cast.cu index 817823c0..a3b10da2 100644 --- a/candle-kernels/src/cast.cu +++ b/candle-kernels/src/cast.cu @@ -28,11 +28,11 @@ extern "C" __global__ void FN_NAME( \ CAST_OP(__nv_bfloat16, __nv_bfloat16, cast_bf16_bf16) CAST_OP(__nv_bfloat16, uint32_t, cast_bf16_u32) -CAST_OP(__nv_bfloat16, __half, cast_bf16_f16) +// CAST_OP(__nv_bfloat16, __half, cast_bf16_f16) CAST_OP(__nv_bfloat16, float, cast_bf16_f32) CAST_OP(__nv_bfloat16, double, cast_bf16_f64) CAST_OP(uint32_t, __nv_bfloat16, cast_u32_bf16) -CAST_OP(__half, __nv_bfloat16, cast_f16_bf16) +// CAST_OP(__half, __nv_bfloat16, cast_f16_bf16) CAST_OP(float, __nv_bfloat16, cast_f32_bf16) CAST_OP(double, __nv_bfloat16, cast_f64_bf16) #endif