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