AlpinDale bfc8988116 feat: add cuda sampling kernels for top_k and top_p (#828) | 1 mēnesi atpakaļ | |
---|---|---|
.. | ||
math.cuh | 1 mēnesi atpakaļ | |
sampling.cu | 1 mēnesi atpakaļ | |
sampling.cuh | 1 mēnesi atpakaļ | |
utils.cuh | 1 mēnesi atpakaļ | |
vec_dtypes.cuh | 1 mēnesi atpakaļ |