Skip to content

Conversation

johnnynunez
Copy link

Support aarch64 including support GH200/GB200
Support Blackwell RTX50/A6000 and B100/B200
Support pytorch 2.7.0
Support cuda 12.8.1

@iofu728 iofu728 requested review from Copilot and iofu728 April 28, 2025 07:12
@iofu728 iofu728 self-assigned this Apr 28, 2025
Copy link

@Copilot Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Copilot encountered an error and was unable to review this pull request. You can try again by re-requesting a review.

@iofu728 iofu728 added the feature feature label Apr 28, 2025
@iofu728
Copy link
Collaborator

iofu728 commented Apr 28, 2025

Hi @johnnynunez, thanks for your help in extending support to Blackwell and ARM architectures.

However, due to two reasons: 1) We currently don't have access to the Blackwell hardware to properly verify correctness, and
2) One part of our code is still CUDA-specific and cannot be directly migrated to ARM yet,

we’re unable to immediately review and merge this PR.
That said, we truly appreciate your contribution and will revisit it once the necessary resources become available. Thanks again!

@johnnynunez
Copy link
Author

Hi @johnnynunez, thanks for your help in extending support to Blackwell and ARM architectures.

However, due to two reasons: 1) We currently don't have access to the Blackwell hardware to properly verify correctness, and 2) One part of our code is still CUDA-specific and cannot be directly migrated to ARM yet,

we’re unable to immediately review and merge this PR. That said, we truly appreciate your contribution and will revisit it once the necessary resources become available. Thanks again!

No problems. You can check it if you want when you have access: https://pypi.jetson-ai-lab.dev/sbsa/cu128

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

feature feature

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants