convert.cuh 391 B

12345678910111213
  1. #include "common.cuh"
  2. #define CUDA_DEQUANTIZE_BLOCK_SIZE 256
  3. template<typename T>
  4. using to_t_cuda_t = void (*)(const void * __restrict__ x, T * __restrict__ y, int64_t k, cudaStream_t stream);
  5. typedef to_t_cuda_t<float> to_fp32_cuda_t;
  6. typedef to_t_cuda_t<half> to_fp16_cuda_t;
  7. to_fp16_cuda_t ggml_get_to_fp16_cuda(ggml_type type);
  8. to_fp32_cuda_t ggml_get_to_fp32_cuda(ggml_type type);