Skip to content

Commit 73b6ea4

Browse files
varun-sundar-rabindranathVarun Sundar Rabindranath
andauthored
support hidden-dim 3072 (#458)
Signed-off-by: Varun Sundar Rabindranath <[email protected]> Co-authored-by: Varun Sundar Rabindranath <[email protected]>
1 parent 4623c67 commit 73b6ea4

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

csrc/kernels/launch.cuh

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -117,6 +117,8 @@
117117
case_macro(2048); \
118118
case 2560: \
119119
case_macro(2560); \
120+
case 3072: \
121+
case_macro(3072); /* for gpt-oss */ \
120122
case 4096: \
121123
case_macro(4096); \
122124
case 5120: \

0 commit comments

Comments
 (0)