Remove torchaudio _check_cuda_version() call when repackaging cu130->cu132

torchaudio is now built against the PyTorch stable C++ ABI, so the
runtime _check_cuda_version() guard in
torchaudio/_extension/__init__.py is no longer needed and would
otherwise fail on cross-CUDA-version installs (e.g. a cu132-tagged
torchaudio wheel running against torch built for a different CUDA
minor version).

Extends release/repackage_torchaudio_cu130_to_cu132.py to also strip
the _check_cuda_version() call from torchaudio/_extension/__init__.py
inside each wheel during the same auditwheel InWheelCtx pass that
rewrites the +cu130 -> +cu132 local-version label, so the regenerated
RECORD covers both edits. The function definition in utils.py is left
intact so external callers (e.g. the pytorch/builder smoke test) keep
working.

Author: atalman

release/repackage_torchaudio_cu130_to_cu132.py

@@ -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,9 @@
 R2_BUCKET_NAME_DEFAULT = "pytorch-downloads"
 R2_PREFIX = "whl/test/{cuda}"
 
+EXTENSION_INIT_PATH = "torchaudio/_extension/__init__.py"
+CHECK_CUDA_CALL_RE = re.compile(r"^[ \t]*_check_cuda_version\(\)[ \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 +188,37 @@ 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 in
+            # torchaudio/_extension/__init__.py 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). Strip the call here; the function
+            # definition in utils.py is left intact for callers like the
+            # builder smoke test.
+            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})"
+            )
+
         new_filename = wheel_path.name.replace(old_label, new_label)
         out_path = output_dir / new_filename
         if not tmp_whl.exists():