Skip to content

Commit

Permalink
Fix RoPE copy-paste mistake
Browse files Browse the repository at this point in the history
  • Loading branch information
sshlyapn committed Aug 1, 2024
1 parent ae3071e commit 93e5a8a
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -73,13 +73,13 @@ KERNEL(rope_ref)(
uint cos_sin_h = h < INPUT1_SIZE_Y ? h : 0;

#ifndef SIN_COS_HAVE_DYNAMIC_PADDINGS
uint cos_sin_idx = INPUT1_GET_INDEX(cos_sin_b, cos_sin_h, cos_sin_p, 0);
uint cos_sin_idx = INPUT1_GET_INDEX(cos_sin_b, cos_sin_p, cos_sin_h, 0);

uint cos_idx = cos_sin_idx;
uint sin_idx = cos_sin_idx;
#else
uint cos_idx = INPUT1_GET_INDEX(cos_sin_b, cos_sin_h, cos_sin_p, 0);
uint sin_idx = INPUT2_GET_INDEX(cos_sin_b, cos_sin_h, cos_sin_p, 0);
uint cos_idx = INPUT1_GET_INDEX(cos_sin_b, cos_sin_p, cos_sin_h, 0);
uint sin_idx = INPUT2_GET_INDEX(cos_sin_b, cos_sin_p, cos_sin_h, 0);
#endif

uint output_idx = OUTPUT_GET_INDEX(b, p, h, 0);
Expand Down

0 comments on commit 93e5a8a

Please sign in to comment.