Skip to content

GitLab

Unverified Commit 022f3cea authored by TJian's avatar TJian Committed by GitHub
Browse files

[ROCm] [Critical]: Remove unused variable (#31156) 


Signed-off-by: default avatartjtanaa <tunjian.tan@embeddedllm.com>
parent a5bc77c2
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
csrc/fused_qknorm_rope_kernel.cu
View file @ 022f3cea
...@@ -405,7 +405,6 @@ void fused_qk_norm_rope( ...@@ -405,7 +405,6 @@ void fused_qk_norm_rope(
qkv.scalar_type() == k_weight.scalar_type(), qkv.scalar_type() == k_weight.scalar_type(),
"qkv, q_weight and k_weight must have the same dtype"); "qkv, q_weight and k_weight must have the same dtype");
int64_t rotary_dim = cos_sin_cache.size(1);
int64_t num_tokens = qkv.size(0); int64_t num_tokens = qkv.size(0);
TORCH_CHECK(position_ids.size(0) == num_tokens, TORCH_CHECK(position_ids.size(0) == num_tokens,
"Number of tokens in position_ids must match QKV"); "Number of tokens in position_ids must match QKV");
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment