ITADN
dmlc/dgl/Issues

[Feature] Support detecting multiple GPUs with different architectures on a single machine.

#7866Pull Requestwanghemeng 创建于 2025-02-09
W
wanghemengcommented
## Description This PR enhances the architecture detection mechanism in DGL by altering the regular expression matching strategy. Specifically, it changes the detection logic from `MATCH` to `MATCHALL`, which enables support for identifying multiple GPUs with different architectures on a single machine, instead of only adapting to the first detected GPU architecture. ## Checklist - ✅ The PR title starts with [Feature]. - ✅ The PR is complete and small. - ✅ All changes have appropriate test coverage. - ✅ The related issue is referenced in this PR. ## Changes - ✅ Updated architecture detection regex from `MATCH` to `MATCHALL`, improving compatibility with multiple GPUs of different architectures.
合并状态:未合并 0 条评论