Does it not support FP8 Kvcache?

#2
by stoneopsx - opened

Does it not support FP8 Kvcache

u needs to modify the "vllm/_custom_ops.py" file to enable support for kv_cache_dtype.

u needs to modify the "vllm/_custom_ops.py" file to enable support for kv_cache_dtype.

could you share how to modify? thanks!

i tried fp8 kv cache on A100 with AI assistant, all backend fail:

  1. FlashAttention: SM80 don't support FP8 attention (require SM90+)
  2. FlashInfer: SM80 don't support trtllm-gen page_size >= 128 (require SM100)
  3. fp8e4nv: SM80 don't support (require SM90+)

i tried fp8 kv cache on A100 with AI assistant, all backend fail:

  1. FlashAttention: SM80 don't support FP8 attention (require SM90+)
  2. FlashInfer: SM80 don't support trtllm-gen page_size >= 128 (require SM100)
  3. fp8e4nv: SM80 don't support (require SM90+)

Thanks for the info. Does it run normally on A100 without FP8 KV cache?

Sign up or log in to comment

Free AI Image Generator No sign-up. Instant results. Open Now