Skip to content

Commit a516003

Browse files
PatriceVignolafacebook-github-bot
authored andcommitted
Remove unused variable in gqa_attn_splitk_attn_kernel (pytorch#4014)
Summary: X-link: facebookresearch/FBGEMM#1102 ATT Reviewed By: helunwencser Differential Revision: D73554136
1 parent 6d64f90 commit a516003

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

fbgemm_gpu/experimental/gen_ai/src/attention/gqa_attn_splitk.cu

-1
Original file line numberDiff line numberDiff line change
@@ -1066,7 +1066,6 @@ __global__ void gqa_attn_splitk_attn_kernel(
10661066
// Each block handles a single batch and head
10671067
auto b = blockIdx.x;
10681068
auto h = blockIdx.y;
1069-
int32_t split_k = XQ_out.size(0);
10701069

10711070
// Note: this is decoding case where we attent to current and all previous
10721071
// tokens.

0 commit comments

Comments
 (0)