Skip to content

[Feature]: DeepSeek-V4 Flash sm_80 (A100/A800) support #40851

Description

@xhd0728

🚀 The feature, motivation and pitch

I am trying to run DeepSeek-V4-Flash on NVIDIA A800 GPUs, which are sm_80 / Ampere GPUs. Currently, using vllm/vllm-openai:deepseekv4-cu129, the model fails during initialization with:

RuntimeError: Assertion error (.../deepgemm-src/csrc/apis/hyperconnection.hpp:56): Unsupported architecture

This still happens with --enforce-eager, so it seems related to the DeepGEMM / HyperConnection kernels rather than torch.compile.

Are there any plans to support sm_80 GPUs such as A100/A800 for DeepSeek-V4, possibly with a slower fallback path? A100/A800 GPUs are still widely used in many research and production clusters, so even a functional fallback would be very helpful.

Alternatives

The current alternatives seem to be:

  1. Use newer GPUs such as H100/H200/B200.
  2. Wait for a backend/kernel fallback that supports sm_80.

I am not sure whether sm_80 support is planned or whether it should currently be considered unsupported for this model.

Additional context

Environment:

  • GPU: NVIDIA A800, sm_80 / Ampere
  • vLLM image: vllm/vllm-openai:deepseekv4-cu129
  • Model: DeepSeek-V4-Flash
  • Single node, 8 GPUs

Simplified command:

docker run -d --gpus all \
  -e CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 \
  -v /path/to/models:/workspace \
  -p 39001:8000 \
  --ipc=host \
  vllm/vllm-openai:deepseekv4-cu129 \
  /workspace/DeepSeek-V4-Flash \
  --served-model-name dsv4-flash \
  --dtype bfloat16 \
  --kv-cache-dtype fp8 \
  --enable-expert-parallel \
  --data-parallel-size 4 \
  --enforce-eager

Before submitting a new issue...

  • Make sure you already searched for relevant issues, and asked the chatbot living at the bottom right corner of the documentation page, which can answer lots of frequently asked questions.

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions