huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 19:47:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add the f16 affine kernel.

Browse Source
This commit is contained in:
laurent
2023-06-26 22:05:31 +01:00
parent 53fdbda683
commit 36a4749e95
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
kernels/src/affine.cu
View File

@ -28,6 +28,10 @@ extern "C" __global__ void FN_NAME( \
} \ } \
} \ } \
#if __CUDA_ARCH__ >= 530
AFFINE_OP(__half, affine_f16)
#endif
AFFINE_OP(float, affine_f32) AFFINE_OP(float, affine_f32)
AFFINE_OP(double, affine_f64) AFFINE_OP(double, affine_f64)
AFFINE_OP(uint32_t, affine_u32) AFFINE_OP(uint32_t, affine_u32)