Skip to content

GitLab

"examples/pytorch/translation/run_translation.py" did not exist on "e6ce636e02ec1cd3f9893af6ab1eec4f113025db"
Switch branch/tag
Find file Normal viewHistoryPermalink
tsembd.cuh 161 Bytes
Newer Older
xuxzh1's avatar
init  
xuxzh1 committed
1
2
3
4
5 
#include "common.cuh"

#define CUDA_TIMESTEP_EMBEDDING_BLOCK_SIZE 256

void ggml_cuda_op_timestep_embedding(ggml_backend_cuda_context & ctx, ggml_tensor * dst);