From 1c1e34735e2d20f20ad17c03f01697ecaae5a8d1 Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Fri, 25 Aug 2023 11:40:36 +0000 Subject: [PATCH] `static_cast` ? --- 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 3e1f5414..0a2282fc 100644 --- a/candle-kernels/src/cast.cu +++ b/candle-kernels/src/cast.cu @@ -13,13 +13,13 @@ extern "C" __global__ void FN_NAME( \ const size_t *strides = info + num_dims; \ if (is_contiguous(num_dims, dims, strides)) { \ for (unsigned int i = blockIdx.x * blockDim.x + threadIdx.x; i < numel; i += blockDim.x * gridDim.x) { \ - out[i] = DST_TYPENAME(inp[i]); \ + out[i] = static_cast(inp[i]); \ } \ } \ else { \ for (unsigned int i = blockIdx.x * blockDim.x + threadIdx.x; i < numel; i += blockDim.x * gridDim.x) { \ unsigned strided_i = get_strided_index(i, num_dims, dims, strides); \ - out[i] = DST_TYPENAME(inp[strided_i]); \ + out[i] = static_cast(inp[strided_i]); \ } \ } \ } \