DeepSpeed
[Hybrid Engine] Enable graph capture for HybridEngine on non-CUDA accelerators
#8933

Merged

[Hybrid Engine] Enable graph capture for HybridEngine on non-CUDA accelerators #8933

delock merged 5 commits into deepspeedai:master from VenusTZZ:npu-hybrid-engine

Make HybridEngine graph capture device-agnostic

553d3592

Fix NPU inference attention for graph capture

fe3dd6d4

Rename hybrid_engine.enable_cuda_graph to enable_graph_capture

454611a4

VenusTZZ marked this pull request as ready for review 13 days ago

VenusTZZ requested a review from

tjruwase 13 days ago

VenusTZZ requested a review from

tohtana 13 days ago

VenusTZZ requested a review from

loadams 13 days ago

Update hybrid engine docs for enable_graph_capture rename

9a4cce35

VenusTZZ force pushed from cf342ec3 to 9a4cce35 13 days ago

delock requested a review from

delock 13 days ago

hipudding requested a review from

hipudding 12 days ago

delock approved these changes on 2026-10-30

0z5a commented on 2026-10-31

Merge branch 'master' into npu-hybrid-engine

acf139d6

delock enabled auto-merge 10 days ago

delock merged ab7fd98f into master 10 days ago

VenusTZZ deleted the npu-hybrid-engine branch 9 days ago

tohtana added needs-gpu-test

Reviewers

delock

0z5a

tjruwase

tohtana

loadams

hipudding

Assignees

No one assigned

Labels

needs-gpu-test

Milestone

No milestone