Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

refactored compute forward to not pass in the src tensors each time #729

Merged
merged 8 commits into from
Feb 21, 2024
Prev Previous commit
Next Next commit
fixed some variable assignments so all tests locally are passing
  • Loading branch information
siddharthvader committed Feb 18, 2024
commit 8e622db9ababb9dad5c34fb59d7cea143566abbd
6 changes: 3 additions & 3 deletions src/ggml.c
Original file line number Diff line number Diff line change
Expand Up @@ -14888,7 +14888,7 @@ static void ggml_compute_forward_map_custom2(
struct ggml_tensor * dst) {

const struct ggml_tensor * a = dst->src[0];
const struct ggml_tensor * b = dst->src[0];
const struct ggml_tensor * b = dst->src[1];

if (params->type == GGML_TASK_INIT || params->type == GGML_TASK_FINALIZE) {
return;
Expand All @@ -14906,8 +14906,8 @@ static void ggml_compute_forward_map_custom3(
struct ggml_tensor * dst) {

const struct ggml_tensor * a = dst->src[0];
const struct ggml_tensor * b = dst->src[0];
const struct ggml_tensor * c = dst->src[0];
const struct ggml_tensor * b = dst->src[1];
const struct ggml_tensor * c = dst->src[2];

if (params->type == GGML_TASK_INIT || params->type == GGML_TASK_FINALIZE) {
return;
Expand Down
Loading