DeepSpeed
[Triton] Make the Triton device guards accelerator-agnostic
#8868
Open
Overview
Commits
12
Changes
View On GitHub

[Triton] Make the Triton device guards accelerator-agnostic #8868

VenusTZZ wants to merge 12 commits into deepspeedai:master from VenusTZZ:npu-triton-support
VenusTZZ
VenusTZZ VenusTZZ requested a review from tjruwase tjruwase 7 days ago
VenusTZZ VenusTZZ requested a review from tohtana tohtana 7 days ago
VenusTZZ VenusTZZ requested a review from hwchen2017 hwchen2017 7 days ago
VenusTZZ VenusTZZ requested a review from loadams loadams 7 days ago
chatgpt-codex-connector
VenusTZZ VenusTZZ marked this pull request as draft 7 days ago
VenusTZZ VenusTZZ force pushed from 77098299 to 5cc7afe9 7 days ago
VenusTZZ VenusTZZ marked this pull request as ready for review 7 days ago
hwchen2017 hwchen2017 requested a review from copilot-pull-request-reviewer copilot-pull-request-reviewer 7 days ago
copilot-pull-request-reviewer
copilot-pull-request-reviewer commented on 2026-10-20
VenusTZZ Make the Triton device guards accelerator-agnostic
9488bd68
VenusTZZ VenusTZZ force pushed from 5cc7afe9 to 9488bd68 7 days ago
jinyouzhi
VenusTZZ Check NPU device availability before opening the Triton gate
37011b0d
VenusTZZ VenusTZZ marked this pull request as draft 6 days ago
VenusTZZ Merge branch 'deepspeedai:master' into npu-triton-support
509605a5
VenusTZZ Gate every Triton kernel call site behind a runtime probe
8985183b
VenusTZZ Harden the Triton kernel probes
d167b4b4
VenusTZZ Add unit tests for the per-kernel Triton probe gate
acfbf62d
VenusTZZ
VenusTZZ Check the accelerator device generator in the probe RNG test
1371124b
jinyouzhi
VenusTZZ
VenusTZZ Draw probe inputs deterministically and harden the probe gate
f803305f
VenusTZZ Answer the Triton gather gate from is_triton_supported
a2a9888a
VenusTZZ VenusTZZ marked this pull request as ready for review 5 days ago
VenusTZZ Dispatch triton_ops kernels through per-hardware subpackages
36e1b910
VenusTZZ VenusTZZ marked this pull request as draft 5 days ago
VenusTZZ Dispatch every training-path Triton kernel entry through an annotation
542efb94
VenusTZZ Add FLA-style Ascend variants for the Triton kernels FLA implements
eb817bca
VenusTZZ VenusTZZ changed the title Make the Triton device guards accelerator-agnostic [Triton] Make the Triton device guards accelerator-agnostic 2 days ago
VenusTZZ VenusTZZ marked this pull request as ready for review 1 day ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum