feat(issue8): MoE combine three-mode precision/traffic trade-off analysis - #16
Open
xyaohubery wants to merge 1 commit into
Open
feat(issue8): MoE combine three-mode precision/traffic trade-off analysis#16xyaohubery wants to merge 1 commit into
xyaohubery wants to merge 1 commit into
Conversation
Analytical model comparing DeepEP V2's three combine reduction modes: - Mode A (no-expand): direct load/store, best precision, minimal SM - Mode B (expand + local reduce): BF16 reduction in shared memory, balanced - Mode C (expanded send): FP32 epilogue reduction, highest traffic, best accuracy Includes: - combine_mode_analysis.py: full analytical model with configurable HW params - test_compare_modes.py: benchmark runner (simulation + real GPU modes) - combine_decision_table.json: pre-computed decision table (231 configs) - README.md: documentation and quick start Co-Authored-By: Claude <noreply@anthropic.com>
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
Analytical model and benchmark framework for DeepEP V2's three combine reduction modes, addressing Issue #8.
Three Modes Analyzed
kUseExpandedLayout=false, kAllowMultipleReduction=falsekUseExpandedLayout=true, kAllowMultipleReduction=truekUseExpandedLayout=true, kAllowMultipleReduction=falseWhat's Included
combine_mode_analysis.py: Analytical model with configurable hardware/workload parameters, based on the actual DeepEP kernel implementationstest_compare_modes.py: Benchmark runner supporting both simulation mode and real GPU hardwarecombine_decision_table.json: Pre-computed decision table across 231 configurations (repetition rate × tokens × hidden dim)README.md: Documentation and quick start guideVerification
Key Finding
For typical MoE workloads (repetition_rate ~0.3), Mode B is recommended:
🤖 Generated with Claude Code