Skip to content

Commit

Permalink
Enable rocm support on 3.13 python (pytorch#6070)
Browse files Browse the repository at this point in the history
Follow up after: pytorch/pytorch#142294
  • Loading branch information
atalman committed Dec 17, 2024
1 parent 00db8ee commit 3617479
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions tools/scripts/generate_binary_build_matrix.py
Original file line number Diff line number Diff line change
Expand Up @@ -469,10 +469,6 @@ def generate_wheels_matrix(
for python_version in python_versions:
for arch_version in arches:

# TODO: Enable Python 3.13 support for ROCM
if arch_version in ROCM_ARCHES and python_version == "3.13":
continue

gpu_arch_type = arch_type(arch_version)
gpu_arch_version = (
"" if arch_version in [CPU, CPU_AARCH64, XPU] else arch_version
Expand Down

0 comments on commit 3617479

Please sign in to comment.