Skip to content

Commit 4356c65

Browse files
committed
feat: add swa_full accessor to LlamaContextParams
1 parent 06c106d commit 4356c65

File tree

1 file changed

+22
-0
lines changed

1 file changed

+22
-0
lines changed

ext/llama_cpp/llama_cpp.c

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -826,6 +826,17 @@ static VALUE llama_context_params_set_op_offload(VALUE self, VALUE op_offload) {
826826
return op_offload;
827827
}
828828

829+
static VALUE llama_context_params_get_swa_full(VALUE self) {
830+
struct llama_context_params* data = get_llama_context_params(self);
831+
return data->swa_full ? Qtrue : Qfalse;
832+
}
833+
834+
static VALUE llama_context_params_set_swa_full(VALUE self, VALUE swa_full) {
835+
struct llama_context_params* data = get_llama_context_params(self);
836+
data->swa_full = RTEST(swa_full) ? true : false;
837+
return swa_full;
838+
}
839+
829840
/* llama_model_quantize_params */
830841
static void llama_model_quantize_params_free(void *ptr) {
831842
if (ptr) {
@@ -4434,6 +4445,17 @@ void Init_llama_cpp(void) {
44344445
* @return [Boolean]
44354446
*/
44364447
rb_define_method(rb_cLlamaContextParams, "op_offload=", RUBY_METHOD_FUNC(llama_context_params_set_op_offload), 1);
4448+
/**
4449+
* Document-method: swa_full
4450+
* @return [Boolean]
4451+
*/
4452+
rb_define_method(rb_cLlamaContextParams, "swa_full", RUBY_METHOD_FUNC(llama_context_params_get_swa_full), 0);
4453+
/**
4454+
* Document-method: swa_full=
4455+
* @param [Boolean] swa_full
4456+
* @return [Boolean]
4457+
*/
4458+
rb_define_method(rb_cLlamaContextParams, "swa_full=", RUBY_METHOD_FUNC(llama_context_params_set_swa_full), 1);
44374459
/* TODO: ggml_abort_callback abort_callback */
44384460
/* TODO: void* abort_callback_data */
44394461

0 commit comments

Comments
 (0)