Skip to content

[Graph Optimization][Speculative Decoding] Update yaml and fix typo#4612

Merged
gongshaotian merged 1 commit intoPaddlePaddle:developfrom
gongshaotian:replace_return_mtp
Oct 28, 2025
Merged

[Graph Optimization][Speculative Decoding] Update yaml and fix typo#4612
gongshaotian merged 1 commit intoPaddlePaddle:developfrom
gongshaotian:replace_return_mtp

Conversation

@gongshaotian
Copy link
Copy Markdown
Collaborator

@gongshaotian gongshaotian commented Oct 27, 2025

Motivation

  • When MTP was enabled for model a, the GPU memory usage fluctuated significantly (67GB to 78GB). Enabling CUDAGraph on this basis resulted in an OOM error, so I adjusted gpu_memory_utilization in the YAML file.
  • Fix typo in spec write cache kernel

Modifications

  1. The value of gpu_memory_utilization in eb45-32k-wint4-mtp-h800-tp4.yaml
  2. spec write cache kernel

Usage or Command

Pass

Accuracy Tests

Pass

Checklist

  • Add at least a tag in the PR title.
    • Tag list: [[FDConfig],[APIServer],[Engine], [Scheduler], [PD Disaggregation], [Executor], [Graph Optimization], [Speculative Decoding], [RL], [Models], [Quantization], [Loader], [OP], [KVCache], [DataProcessor], [BugFix], [Docs], [CI], [Optimization], [Feature], [Benchmark], [Others], [XPU], [HPU], [GCU], [DCU], [Iluvatar], [Metax]]
    • You can add new tags based on the PR content, but the semantics must be clear.
  • Format your code, run pre-commit before commit.
  • Add unit tests. Please write the reason in this PR if no unit tests.
  • Provide accuracy results.
  • If the current PR is submitting to the release branch, make sure the PR has been submitted to the develop branch, then cherry-pick it to the release branch with the [Cherry-Pick] PR tag.

@paddle-bot
Copy link
Copy Markdown

paddle-bot bot commented Oct 27, 2025

Thanks for your contribution!

@gongshaotian gongshaotian merged commit 86d5006 into PaddlePaddle:develop Oct 28, 2025
26 of 27 checks passed
@gongshaotian gongshaotian deleted the replace_return_mtp branch November 3, 2025 06:49
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants