Skip to content

Commit cbb0811

Browse files
authored
del append group-size=24 (#8048)
1 parent 4ba6625 commit cbb0811

2 files changed

Lines changed: 3 additions & 6 deletions

File tree

custom_ops/gpu_ops/append_attn/template_config.json

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
"IsDynamicC8"
1818
],
1919
"dispatch_params": {
20-
"GROUP_SIZE": [1, 2, 3, 4, 5, 6, 7, 8, 12, 14, 16, 24],
20+
"GROUP_SIZE": [1, 2, 3, 4, 5, 6, 7, 8, 12, 14, 16],
2121
"HEAD_DIM": [128],
2222
"BLOCK_SIZE": [64],
2323
"CAUSAL": [0, 1],
@@ -54,7 +54,7 @@
5454
"ENABLE_PREFILL"
5555
],
5656
"dispatch_params": {
57-
"GROUP_SIZE": [1, 2, 3, 4, 5, 6, 7, 8, 12, 14, 16, 24],
57+
"GROUP_SIZE": [1, 2, 3, 4, 5, 6, 7, 8, 12, 14, 16],
5858
"HEAD_DIM": [128],
5959
"BLOCK_SIZE": [64],
6060
"CAUSAL": [0, 1],
@@ -89,7 +89,7 @@
8989
"ENABLE_PREFILL"
9090
],
9191
"dispatch_params": {
92-
"GROUP_SIZE": [1, 2, 3, 4, 5, 6, 7, 8, 12, 14, 16, 24],
92+
"GROUP_SIZE": [1, 2, 3, 4, 5, 6, 7, 8, 12, 14, 16],
9393
"HEAD_DIM": [64,128,192],
9494
"BLOCK_SIZE": [64],
9595
"CAUSAL": [0, 1],

custom_ops/gpu_ops/append_attn/utils.cuh

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -450,9 +450,6 @@ __forceinline__ __host__ __device__ void vec_cast<nv_bfloat16, float>(
450450
} else if (group_size == 16) { \
451451
constexpr size_t GROUP_SIZE = 16; \
452452
__VA_ARGS__ \
453-
} else if (group_size == 24) { \
454-
constexpr size_t GROUP_SIZE = 24; \
455-
__VA_ARGS__ \
456453
} else { \
457454
PD_THROW("not support the group_size", group_size); \
458455
}

0 commit comments

Comments
 (0)