Skip to content

Commit f9751b5

Browse files
xiao-llmamd-xiaoyu12
authored andcommitted
Update _custom_ops.py
Signed-off-by: Xiao Yu <[email protected]>
1 parent 6f488d8 commit f9751b5

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

vllm/_custom_ops.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,6 @@ def paged_attention_rocm(
119119
fp8_out_scale: Optional[torch.Tensor] = None,
120120
mfma_type: str = "fp8" if envs.is_set("VLLM_USE_FP8_MFMA") else "f16",
121121
) -> None:
122-
print(f"\nmfma_type: {mfma_type}")
123122
torch.ops._rocm_C.paged_attention(out, exp_sum, max_logits, tmp_out, query,
124123
key_cache, value_cache, num_kv_heads,
125124
scale, block_tables, seq_lens,

0 commit comments

Comments
 (0)