Skip to content

Commit

Permalink
Update llama.cpp
Browse files Browse the repository at this point in the history
  • Loading branch information
LLukas22 authored and AmineDiro committed Aug 15, 2023
1 parent a8398a6 commit b77388e
Show file tree
Hide file tree
Showing 3 changed files with 16 additions and 1 deletion.
3 changes: 3 additions & 0 deletions crates/ggml/sys/src/cuda.rs
Original file line number Diff line number Diff line change
Expand Up @@ -60,6 +60,9 @@ extern "C" {
extern "C" {
pub fn ggml_cuda_set_main_device(main_device: ::std::os::raw::c_int);
}
extern "C" {
pub fn ggml_cuda_set_mul_mat_q(mul_mat_q: bool);
}
extern "C" {
pub fn ggml_cuda_set_scratch_size(scratch_size: usize);
}
Expand Down
12 changes: 12 additions & 0 deletions crates/ggml/sys/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -1567,6 +1567,18 @@ extern "C" {
n_ctx: ::std::os::raw::c_int,
) -> *mut ggml_tensor;
}
extern "C" {
pub fn ggml_rope_custom(
ctx: *mut ggml_context,
a: *mut ggml_tensor,
n_past: ::std::os::raw::c_int,
n_dims: ::std::os::raw::c_int,
mode: ::std::os::raw::c_int,
n_ctx: ::std::os::raw::c_int,
freq_base: f32,
freq_scale: f32,
) -> *mut ggml_tensor;
}
extern "C" {
pub fn ggml_rope_custom_inplace(
ctx: *mut ggml_context,
Expand Down

0 comments on commit b77388e

Please sign in to comment.