Skip to content

[Rocm6.4] Update triton requirements and conditionalize python_version < '3.13' only for PyTorch <= 2.5 (#68) AND add in conda to dockerfile for torchvision#70

Merged
jithunnair-amd merged 4 commits intorocm6.4from
rocm6.4_torch_commit
Feb 17, 2025

Conversation

@ethanwee1
Copy link

@ethanwee1 ethanwee1 commented Feb 13, 2025

Required for torchvision 0.21 build for PyTorch 2.6
(relates to ROCm/pytorch#1884)
Brings in name conditional changes for triton wheels also.

Validation:
http://ml-ci-internal.amd.com:8080/job/pytorch/job/manylinux_rocm_wheels/553/

* Append torch commit if it is set

* Remove git

* Rename wheels

* lint

* refactor
…n < '3.13' only for PyTorch <= 2.5 (#68)

* conditionalize and python_version < '3.13'" only for PyTorch 2.5 and older

* Limit TRITON_CONSTRAINT

* Change triton wheel name to .git with 8 characters
@ethanwee1 ethanwee1 changed the title [Rocm6.4] Update triton requirements and conditionalize python_version < '3.13' only for PyTorch <= 2.5 (#68) [Rocm6.4] Update triton requirements and conditionalize python_version < '3.13' only for PyTorch <= 2.5 (#68) AND add in conda to dockerfile for torchvision Feb 15, 2025
@jithunnair-amd jithunnair-amd merged commit 2b26379 into rocm6.4 Feb 17, 2025
1 check failed
@ethanwee1 ethanwee1 deleted the rocm6.4_torch_commit branch March 4, 2025 18:51
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