Skip to content

Conversation

manoelmarques
Copy link
Contributor

In preprocess scritp, set env. TORCH_CUDA_ARCH_LIST with the gpu capability of the gpu to be used in order to speed pytorch compilation.

@manoelmarques manoelmarques self-assigned this Oct 10, 2025
@manoelmarques manoelmarques merged commit f157e49 into llm-d:main Oct 13, 2025
8 checks passed
@manoelmarques manoelmarques deleted the gpu branch October 13, 2025 14:38
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants