fix(musa): expose device index helper for TorchDynamo

README.md

@@ -392,7 +392,7 @@ See `src/torchada/_mappings/` for 400+ mapping rules grouped by API domain.
 
 ```
 # pyproject.toml or requirements.txt
-torchada>=0.1.85
+torchada>=0.1.86
 ```
 
 ### Step 2: Conditional Import

README_CN.md

@@ -376,7 +376,7 @@ if torchada.is_gpu_device(device):  # 在 CUDA 和 MUSA 上都能工作
 
 ```
 # pyproject.toml 或 requirements.txt
-torchada>=0.1.85
+torchada>=0.1.86
 ```
 
 ### 步骤 2：条件导入

benchmarks/benchmark_history.json

@@ -3,7 +3,7 @@
   "description": "Historical benchmark results for torchada performance tracking",
   "results": [
     {
-      "version": "0.1.85",
+      "version": "0.1.86",
       "date": "2026-01-29",
       "platform": "MUSA",
       "pytorch_version": "2.7.1",

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "torchada"
-version = "0.1.85"
+version = "0.1.86"
 description = "Adapter package for torch_musa to act exactly like PyTorch CUDA"
 readme = "README.md"
 license = {text = "MIT"}

src/torchada/__init__.py

@@ -24,7 +24,7 @@
     from torch.utils.cpp_extension import CUDAExtension, BuildExtension, CUDA_HOME
 """
 
-__version__ = "0.1.85"
+__version__ = "0.1.86"
 
 from . import cuda, utils
 

src/torchada/_patch.py

@@ -827,6 +827,7 @@ class _CudaModuleWrapper(ModuleType):
     _SPECIAL_ATTRS = {
         "StreamContext": "core.stream.StreamContext",
         "streams": "core.stream",
+        "_get_device_index": "core._utils._get_musa_device_index",
     }
 
     # Attribute name remappings (CUDA name -> MUSA name)

tests/test_cuda_patching.py

@@ -729,6 +729,17 @@ def test_streams_module(self):
             assert sys.modules["torch.cuda.streams"] is torch_musa.core.stream
             assert torch.cuda.streams.Stream is torch_musa.core.stream.Stream
 
+    def test_get_device_index_helper(self):
+        """Test the private CUDA device helper used by PyTorch Dynamo."""
+        import torch
+
+        import torchada
+
+        if torchada.is_musa_platform():
+            from torch_musa.core._utils import _get_musa_device_index
+
+            assert torch.cuda._get_device_index is _get_musa_device_index
+
     def test_stream_cuda_stream_property(self):
         """Test stream.cuda_stream returns same value as stream.musa_stream on MUSA."""
         import torch

tests/test_platform.py

@@ -59,7 +59,7 @@ def test_get_version(self):
 
         version = torchada.get_version()
         assert version == torchada.__version__
-        assert version == "0.1.85"
+        assert version == "0.1.86"
         assert isinstance(version, str)
 
     def test_project_version_matches_runtime_version(self):