Skip to content

Commit 726704a

Browse files
authored
feat: Support d_conv=15 for ssm-conv.cu (ggml-org#23017)
Branch: ModalityConditionalAdapters AI-usage: none Signed-off-by: Gabe Goodhart <ghart@us.ibm.com>
1 parent 8758904 commit 726704a

1 file changed

Lines changed: 6 additions & 5 deletions

File tree

ggml/src/ggml-cuda/ssm-conv.cu

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -140,11 +140,12 @@ static void ssm_conv_f32_cuda(const float * src0, const float * src1, const floa
140140
};
141141

142142
switch (nc) {
143-
case 3: launch_kernel(std::integral_constant<int, 3>{}); break;
144-
case 4: launch_kernel(std::integral_constant<int, 4>{}); break;
145-
case 5: launch_kernel(std::integral_constant<int, 5>{}); break;
146-
case 9: launch_kernel(std::integral_constant<int, 9>{}); break;
147-
default: GGML_ABORT("Only support kernel sizes 3, 4, 5, 9 right now.");
143+
case 3: launch_kernel(std::integral_constant<int, 3 >{}); break;
144+
case 4: launch_kernel(std::integral_constant<int, 4 >{}); break;
145+
case 5: launch_kernel(std::integral_constant<int, 5 >{}); break;
146+
case 9: launch_kernel(std::integral_constant<int, 9 >{}); break;
147+
case 15: launch_kernel(std::integral_constant<int, 15>{}); break;
148+
default: GGML_ABORT("Only support kernel sizes 3, 4, 5, 9, 15 right now.");
148149
}
149150
}
150151

0 commit comments

Comments
 (0)