pytorch
diff --git a/‎release/repackage_torchaudio_cu130_to_cu132.py‎
Lines changed: 85 additions & 1 deletion b/‎release/repackage_torchaudio_cu130_to_cu132.py‎
Lines changed: 85 additions & 1 deletion
@@ -9,7 +9,15 @@
   3. Rewrites the wheel local-version label from ``+cu130`` to ``+cu132``
      (METADATA ``Version:`` field, ``.dist-info`` directory name, and the
      wheel filename). The RECORD file is regenerated by auditwheel.
-  4. Uploads each repackaged wheel to:
+  4. Removes the ``_check_cuda_version()`` runtime guard from
+     ``torchaudio/_extension/__init__.py`` inside the wheel. torchaudio is
+     now built against the PyTorch stable C++ ABI, so the strict CUDA
+     version match is no longer required and would otherwise fail when a
+     cu132-tagged torchaudio wheel runs alongside a torch built for a
+     different CUDA minor version. The function definition in ``utils.py``
+     is preserved so external callers (e.g. the pytorch/builder smoke
+     test) keep working.
+  5. Uploads each repackaged wheel to:
        - AWS S3:  s3://pytorch/whl/test/cu132/
        - Cloudflare R2: s3://pytorch-downloads/whl/test/cu132/
      Each upload sets ``x-amz-meta-checksum-sha256`` on the object so the
@@ -56,6 +64,18 @@
 R2_BUCKET_NAME_DEFAULT = "pytorch-downloads"
 R2_PREFIX = "whl/test/{cuda}"
 
+EXTENSION_INIT_PATH = "torchaudio/_extension/__init__.py"
+EXTENSION_UTILS_PATH = "torchaudio/_extension/utils.py"
+CHECK_CUDA_CALL_RE = re.compile(r"^[ \t]*_check_cuda_version\(\)[ \t]*\n", re.MULTILINE)
+# Match the `if ta_version != t_version: raise RuntimeError(...)` block inside
+# _check_cuda_version() in utils.py. The raise spans multiple lines, so we
+# match the whole if-body by indent.
+CHECK_CUDA_RAISE_RE = re.compile(
+    r"^(?P<indent>[ \t]+)if ta_version != t_version:[ \t]*\n"
+    r"(?:(?P=indent)[ \t]+.*\n)+",
+    re.MULTILINE,
+)
+
 
 def discover_wheels(package: str, version: str, cuda: str) -> List[str]:
     """Return list of wheel filenames for ``package`` at ``version`` on the
@@ -177,6 +197,70 @@ def repackage_wheel(
                     os.path.join(ctx.path, new_dist_info_dir),
                 )
 
+            # torchaudio is now built against the PyTorch stable C++ ABI,
+            # so the runtime _check_cuda_version() guard is no longer needed
+            # and would otherwise fail on cross-CUDA-version installs (e.g.
+            # a torchaudio cu132 wheel running against torch built for a
+            # different CUDA minor). Two patches are required:
+            #   (a) strip the call site in __init__.py so importing the
+            #       package does not raise, and
+            #   (b) neutralize the version-mismatch raise in utils.py so
+            #       external callers (e.g. the pytorch smoke test, which
+            #       calls torchaudio._extension._check_cuda_version()
+            #       directly) do not raise either.
+            init_file = os.path.join(ctx.path, EXTENSION_INIT_PATH)
+            if not os.path.exists(init_file):
+                raise RuntimeError(f"{EXTENSION_INIT_PATH} not found in {wheel_path}")
+            with open(init_file, "r", encoding="utf-8") as f:
+                init_src = f.read()
+            patched_src, count = CHECK_CUDA_CALL_RE.subn("", init_src)
+            if count == 0:
+                raise RuntimeError(
+                    f"No _check_cuda_version() call found in "
+                    f"{EXTENSION_INIT_PATH} of {wheel_path.name}"
+                )
+            if count > 1:
+                print(
+                    f"- WARNING: removed {count} _check_cuda_version() "
+                    f"calls from {wheel_path.name} (expected 1)"
+                )
+            with open(init_file, "w", encoding="utf-8") as f:
+                f.write(patched_src)
+            print(
+                f"+ Removed _check_cuda_version() call from "
+                f"{EXTENSION_INIT_PATH} ({wheel_path.name})"
+            )
+
+            utils_file = os.path.join(ctx.path, EXTENSION_UTILS_PATH)
+            if not os.path.exists(utils_file):
+                raise RuntimeError(f"{EXTENSION_UTILS_PATH} not found in {wheel_path}")
+            with open(utils_file, "r", encoding="utf-8") as f:
+                utils_src = f.read()
+            patched_utils, utils_count = CHECK_CUDA_RAISE_RE.subn(
+                lambda m: (
+                    f"{m.group('indent')}pass  "
+                    f"# CUDA version mismatch check disabled by "
+                    f"repackage_torchaudio_cu130_to_cu132.py\n"
+                ),
+                utils_src,
+            )
+            if utils_count == 0:
+                raise RuntimeError(
+                    f"No `if ta_version != t_version:` block found in "
+                    f"{EXTENSION_UTILS_PATH} of {wheel_path.name}"
+                )
+            if utils_count > 1:
+                print(
+                    f"- WARNING: replaced {utils_count} CUDA-mismatch "
+                    f"raise blocks in {wheel_path.name} (expected 1)"
+                )
+            with open(utils_file, "w", encoding="utf-8") as f:
+                f.write(patched_utils)
+            print(
+                f"+ Disabled CUDA version mismatch raise in "
+                f"{EXTENSION_UTILS_PATH} ({wheel_path.name})"
+            )
+
         new_filename = wheel_path.name.replace(old_label, new_label)
         out_path = output_dir / new_filename
         if not tmp_whl.exists():