DeepSpeed
Avoid CUDA context initialization during op compatibility checks at import
#8078

Merged

Avoid CUDA context initialization during op compatibility checks at import #8078

tohtana merged 4 commits into deepspeedai:master from Achyuthan-S:fix/import-fork-safety

Fix fork poisoning from import-time CUDA capability probes (#7918)

02b1c335

Achyuthan-S requested a review from

copilot-pull-request-reviewer 148 days ago

Achyuthan-S requested a review from

tjruwase 148 days ago

Achyuthan-S requested a review from

loadams 148 days ago

Achyuthan-S requested a review from

tohtana 148 days ago

copilot-pull-request-reviewer commented on 2026-06-19

chatgpt-codex-connector commented on 2026-06-19

Make import fork-safe via NVML-based CUDA availability check

88550889

Don't probe device capability at import in ds_transformer

c188bd1f

Merge upstream master into fix/import-fork-safety

fc9757f8

tohtana approved these changes on 2026-06-29

tohtana enabled auto-merge 138 days ago

tohtana merged 5aef6d88 into master 138 days ago

Reviewers

tohtana

tjruwase

loadams

Assignees

No one assigned

Labels

None yet

Milestone

No milestone