Skip to content

Commit

Permalink
fix affine if >= 530 < 800
Browse files Browse the repository at this point in the history
  • Loading branch information
haricot committed Jan 23, 2025
1 parent 09e3d0b commit 33dcb9a
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion candle-kernels/src/affine.cu
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down

0 comments on commit 33dcb9a

Please sign in to comment.