mmq.hpp 819 B

123456789101112131415161718192021222324252627282930313233
  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_MMQ_HPP
  12. #define GGML_SYCL_MMQ_HPP
  13. #include "common.hpp"
  14. void ggml_sycl_op_mul_mat_q(
  15. ggml_backend_sycl_context & ctx,
  16. const ggml_tensor* src0,
  17. const ggml_tensor* src1,
  18. ggml_tensor* dst,
  19. const char* src0_dd_i,
  20. const float* src1_ddf_i,
  21. const char* src1_ddq_i,
  22. float* dst_dd_i,
  23. const int64_t row_low,
  24. const int64_t row_high,
  25. const int64_t src1_ncols,
  26. const int64_t src1_padded_row_size,
  27. const dpct::queue_ptr& stream);
  28. #endif // GGML_SYCL_MMQ_HPP