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

pesky bfloat type

Browse Source
This commit is contained in:
Ivar Flakstad
2023-11-07 10:26:59 +01:00
parent 7ff17d92b3
commit 1367e0278b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-metal-kernels/src/unary.metal
View File

@ -37,7 +37,7 @@ kernel void FN_NAME( \
const size_t start = thread_index * length; \ const size_t start = thread_index * length; \
const size_t stop = min(start + length, dim); \ const size_t stop = min(start + length, dim); \
for (size_t i = start; i < stop; i++){ \ for (size_t i = start; i < stop; i++){ \
output[i] = FN(input[i]); \ output[i] = TYPENAME(FN(input[i])); \
} \ } \
}\ }\
kernel void FN_NAME_STRIDED( \ kernel void FN_NAME_STRIDED( \
@ -55,7 +55,7 @@ kernel void FN_NAME_STRIDED( \
const size_t start = thread_index * length; \ const size_t start = thread_index * length; \
const size_t stop = min(start + length, dim); \ const size_t stop = min(start + length, dim); \
for (size_t i = start; i < stop; i++){ \ for (size_t i = start; i < stop; i++){ \
output[i] = FN(input[get_strided_index(i, num_dims, dims, strides, offset)]); \ output[i] = TYPENAME(FN(input[get_strided_index(i, num_dims, dims, strides, offset)])); \
} \ } \
} }