diff options
| author | laurent <laurent.mazare@gmail.com> | 2023-06-29 23:23:44 +0100 |
|---|---|---|
| committer | laurent <laurent.mazare@gmail.com> | 2023-06-29 23:23:44 +0100 |
| commit | 6486a6d7b29b6f89a8a23a696cf6614ac14f0b36 (patch) | |
| tree | e38f7642e57897a6f6d955fd811ac36dcb5abc33 /candle-kernels | |
| parent | ec79fc43f21bd0c50053ea9b5b66104464cdcb66 (diff) | |
| download | candle-6486a6d7b29b6f89a8a23a696cf6614ac14f0b36.tar.gz candle-6486a6d7b29b6f89a8a23a696cf6614ac14f0b36.tar.bz2 candle-6486a6d7b29b6f89a8a23a696cf6614ac14f0b36.zip | |
Avoid some cast kernels.
Diffstat (limited to 'candle-kernels')
| -rw-r--r-- | candle-kernels/src/cast.cu | 4 |
1 files 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 |