Skip to content

Commit 62b09b3

Browse files
committed
metal : fix wrong number of tokens per sequence in SSM_SCAN
1 parent 5b8ec2b commit 62b09b3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ggml/src/ggml-metal.m

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1690,7 +1690,7 @@ static void ggml_metal_encode_node(
16901690
const int64_t d_inner = ne01;
16911691
const int64_t n_head = ne02;
16921692
const int64_t n_group = ne41;
1693-
const int64_t n_seq_tokens = ne11;
1693+
const int64_t n_seq_tokens = ne12;
16941694
const int64_t n_seqs = ne13;
16951695

16961696
id<MTLComputePipelineState> pipeline = nil;

0 commit comments

Comments
 (0)