deepseek-ai/DeepEP
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepEP.git synced 2026-11-05 00:47:32 +00:00
Code Issues 308 Projects Packages Wiki Activity

[PR #167] [MERGED] Make TORCH_CUDA_ARCH_LIST as an environment variable #446

New issue
Closed
opened 2026-10-26 10:00:51 +00:00 by bsakaguchi · 0 comments
bsakaguchi commented 2026-10-26 10:00:51 +00:00
Owner
Copy link

📋 Pull Request Information

Original PR: https://github.com/deepseek-ai/DeepEP/pull/167
Author: @guyueh1
Created: 5/14/2025
Status: ✅ Merged
Merged: 5/19/2025
Merged by: @LyricZhao

Base: main ← Head: main

📝 Commits (2)

  • 1c91155 Add 10.0 to TORCH_CUDA_ARCH_LIST
  • 64285ba Revert csrc/CMakeLists change; in setup.py make TORCH_CUDA_ARCH_LIST configurable

📊 Changes

1 file changed (+2 additions, -1 deletions)

View changed files

📝 setup.py (+2 -1)

📄 Description

This PR adds the 10.0 to TORCH_CUDA_ARCH_LIST to support compilation for Blackwell GPUs.

🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.

## 📋 Pull Request Information **Original PR:** https://github.com/deepseek-ai/DeepEP/pull/167 **Author:** [@guyueh1](https://github.com/guyueh1) **Created:** 5/14/2025 **Status:** ✅ Merged **Merged:** 5/19/2025 **Merged by:** [@LyricZhao](https://github.com/LyricZhao) **Base:** `main` ← **Head:** `main` --- ### 📝 Commits (2) - [`1c91155`](https://github.com/deepseek-ai/DeepEP/commit/1c91155365d6893ddab834523dfdfbdfc68a8794) Add 10.0 to TORCH_CUDA_ARCH_LIST - [`64285ba`](https://github.com/deepseek-ai/DeepEP/commit/64285bad4fd543efb09ea6f69c97dbf795d5117f) Revert csrc/CMakeLists change; in setup.py make TORCH_CUDA_ARCH_LIST configurable ### 📊 Changes **1 file changed** (+2 additions, -1 deletions) <details> <summary>View changed files</summary> 📝 `setup.py` (+2 -1) </details> ### 📄 Description This PR adds the 10.0 to TORCH_CUDA_ARCH_LIST to support compilation for Blackwell GPUs. --- <sub>🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.</sub>
bsakaguchi 2026-10-26 10:00:51 +00:00
Sign in to join this conversation.
No Branch/Tag specified
Branches Tags
main
hybrid-ep
chore/remove-clang-format-yapf
v2.5
nvDev
try-fix-cu128
mori-ep
antgroup-opt
compat-old-torch
add-linter
eager-rdma
v1.2.1
Labels
Clear labels   
pull-request
Mirrored from GitHub Pull Request

No labels
pull-request
Milestone
Clear milestone
No items
No milestone
Projects
Clear projects
No items
No project
Assignees
Clear assignees
No assignees
1 participant
Notifications
Due date
The due date is invalid or out of range. Please use the format "yyyy-mm-dd".

No due date set.

Dependencies

No dependencies set.

Reference
deepseek-ai/DeepEP#446
No description provided.