From 33dcb9aad4cf6e965f3addb281e1bc5f3eebfb71 Mon Sep 17 00:00:00 2001 From: Nicolas <344493+haricot@users.noreply.github.com> Date: Thu, 23 Jan 2025 17:02:10 +0100 Subject: [PATCH] fix affine if >= 530 < 800 --- candle-kernels/src/affine.cu | 1 - 1 file changed, 1 deletion(-) diff --git a/candle-kernels/src/affine.cu b/candle-kernels/src/affine.cu index c1ee5e140..ba86abe32 100644 --- a/candle-kernels/src/affine.cu +++ b/candle-kernels/src/affine.cu @@ -33,7 +33,6 @@ AFFINE_OP(__nv_bfloat16, affine_bf16) #endif #if __CUDA_ARCH__ >= 530 -AFFINE_OP(__nv_bfloat16, affine_bf16) AFFINE_OP(__half, affine_f16) #endif