dmmv.hpp 808 B

123456789101112131415161718192021222324252627
  1. //
  2. // MIT license
  3. // Copyright (C) 2024 Intel Corporation
  4. // SPDX-License-Identifier: MIT
  5. //
  6. //
  7. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
  8. // See https://llvm.org/LICENSE.txt for license information.
  9. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
  10. //
  11. #ifndef GGML_SYCL_DMMV_HPP
  12. #define GGML_SYCL_DMMV_HPP
  13. #include "common.hpp"
  14. void ggml_sycl_op_dequantize_mul_mat_vec(
  15. ggml_backend_sycl_context & ctx,
  16. const ggml_tensor *src0, const ggml_tensor *src1, ggml_tensor *dst,
  17. const char *src0_dd_i, const float *src1_ddf_i, const char *src1_ddq_i,
  18. float *dst_dd_i, const int64_t row_low, const int64_t row_high,
  19. const int64_t src1_ncols, const int64_t src1_padded_row_size,
  20. const dpct::queue_ptr &stream);
  21. #endif // GGML_SYCL_DMMV_HPP