Skip to content

Commit

Permalink
Fix input indexes
Browse files Browse the repository at this point in the history
  • Loading branch information
sshlyapn committed Aug 1, 2024
1 parent 4f06119 commit 20cfc1b
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@ JitConstants RoPEKernelBase::GetJitConstants(const rope_params& params, RoPEKern
jit.AddConstant(MakeJitConstant("TRANSPOSED_INPUT0_BATCH_PITCH", "INPUT0_BATCH_PITCH"));
}

if (params.inputs[0].has_dynamic_pad() || params.inputs[1].has_dynamic_pad()) {
if (!params.is_chatglm && (params.inputs[1].has_dynamic_pad() || params.inputs[2].has_dynamic_pad())) {
jit.AddConstant(MakeJitConstant("SIN_COS_HAVE_DYNAMIC_PADDINGS", true));
}

Expand Down

0 comments on commit 20cfc1b

Please sign in to comment.