ggml-metal: fix custom rope

This commit is contained in:
Xiao-Yong Jin 2023-06-30 00:45:34 -05:00
parent dc0d0eb6a9
commit 1ae4318ddb

View file

@ -876,7 +876,7 @@ void ggml_metal_graph_compute(
float freq_base, freq_scale;
memcpy(&freq_base, (int32_t *) src1->data + 4, sizeof(float));
memcpy(&freq_scale, (int32_t *) src1->date + 5, sizeof(float));
memcpy(&freq_scale, (int32_t *) src1->data + 5, sizeof(float));
[encoder setComputePipelineState:ctx->pipeline_rope];
[encoder setBuffer:id_src0 offset:offs_src0 atIndex:0];