Skip to content

Conversation

@Hamerlate
Copy link
Contributor

The base image rocm/pytorch:rocm6.3.2_ubuntu22.04_py3.10_pytorch_release_2.4.0 already includes conda.

The AMD-specific PyTorch environment is located in the py_3.10 conda environment. This PR updates the Docker startup commands accordingly to activate the correct environment.

Tested successfully on MI300X hardware.

@github-actions
Copy link

👋 Hi! Thank you for contributing to the TileLang project.

Please remember to run bash format.sh in the root directory of the project to ensure your changes are properly linted and formatted. This will help ensure your contribution passes the format check.

We appreciate you taking this step! Our team will review your contribution, and we look forward to your awesome work!

🚀

@LeiWang1999
Copy link
Member

LGTM, Merged!

@LeiWang1999 LeiWang1999 merged commit 30ca6da into tile-ai:main May 20, 2025
3 checks passed
LeiWang1999 pushed a commit to LeiWang1999/tilelang that referenced this pull request Jul 18, 2025
Co-authored-by: Ubuntu <srguser@srgmi300c.ibcr0fi0qgdu5pqgbnhfbyasxg.parx.internal.cloudapp.net>
LeiWang1999 pushed a commit to LeiWang1999/tilelang that referenced this pull request Jul 20, 2025
Co-authored-by: Ubuntu <srguser@srgmi300c.ibcr0fi0qgdu5pqgbnhfbyasxg.parx.internal.cloudapp.net>
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