| 1 | #include "common.cuh" |
|---|---|
| 2 | |
| 3 | #define CUDA_TIMESTEP_EMBEDDING_BLOCK_SIZE 256 |
| 4 | |
| 5 | void ggml_cuda_op_timestep_embedding(ggml_backend_cuda_context & ctx, ggml_tensor * dst); |
| 6 |
| 1 | #include "common.cuh" |
|---|---|
| 2 | |
| 3 | #define CUDA_TIMESTEP_EMBEDDING_BLOCK_SIZE 256 |
| 4 | |
| 5 | void ggml_cuda_op_timestep_embedding(ggml_backend_cuda_context & ctx, ggml_tensor * dst); |
| 6 |