Build (aarch64-linux)

build/torch26-cxx11-cu126-aarch64-linux/paged_attention/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _paged_attention_daf6221
-ops = torch.ops._paged_attention_daf6221
+from . import _paged_attention_6677800
+ops = torch.ops._paged_attention_6677800
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_paged_attention_daf6221::{op_name}"
+    return f"_paged_attention_6677800::{op_name}"

build/torch26-cxx11-cu126-aarch64-linux/paged_attention/{_paged_attention_daf6221.abi3.so → _paged_attention_6677800.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ac7b8c17999df02a6869772e37713c22d3a765747daa576e64a4ff84a246dfaa
-size 87576024
+oid sha256:4b860f51a86eb13c6c7a6e74f28c59abf0b73245f241744766029fbda53cba87
+size 88169704

build/torch26-cxx11-cu126-aarch64-linux/paged_attention/platforms.py

@@ -8,6 +8,7 @@ import numpy as np
 import torch
 
 IS_ROCM = torch.version.hip is not None
+IS_MPS = torch.backends.mps.is_available()
 
 
 class Platform(ABC):
@@ -32,6 +33,9 @@ class Platform(ABC):
     @abstractmethod
     def is_rocm(self) -> bool: ...
 
+    @abstractmethod
+    def is_mps(self) -> bool: ...
+
 
 class CudaPlatform(Platform):
     @classmethod
@@ -45,6 +49,9 @@ class CudaPlatform(Platform):
     def is_rocm(self) -> bool:
         return False
 
+    def is_mps(self) -> bool:
+        return False
+
 
 class RocmPlatform(Platform):
     @classmethod
@@ -58,5 +65,28 @@ class RocmPlatform(Platform):
     def is_rocm(self) -> bool:
         return True
 
+    def is_mps(self) -> bool:
+        return False
+
+
+class MpsPlatform(Platform):
+    @classmethod
+    @lru_cache(maxsize=8)
+    def get_device_name(cls, device_id: int = 0) -> str:
+        return torch.cuda.get_device_name(device_id)
+
+    def is_cuda(self) -> bool:
+        return False
+
+    def is_rocm(self) -> bool:
+        return False
+
+    def is_mps(self) -> bool:
+        return True
 
-current_platform = RocmPlatform() if IS_ROCM else CudaPlatform()
+current_platform = (
+    RocmPlatform() if IS_ROCM else
+    MpsPlatform() if IS_MPS else
+    CudaPlatform() if torch.cuda.is_available() else
+    None
+)

build/torch26-cxx98-cu126-aarch64-linux/paged_attention/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _paged_attention_daf6221
-ops = torch.ops._paged_attention_daf6221
+from . import _paged_attention_6677800
+ops = torch.ops._paged_attention_6677800
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_paged_attention_daf6221::{op_name}"
+    return f"_paged_attention_6677800::{op_name}"

build/torch26-cxx98-cu126-aarch64-linux/paged_attention/{_paged_attention_daf6221.abi3.so → _paged_attention_6677800.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:854c53fdffb1ee8f889cbbb78127cf92a843c55d00228b8aadcd48050d700af0
-size 87568816
+oid sha256:385c8fd6816793573ac802c40e1be99cc5703eebb4df05330ce57b2216da8836
+size 88162432

build/torch26-cxx98-cu126-aarch64-linux/paged_attention/platforms.py

@@ -8,6 +8,7 @@ import numpy as np
 import torch
 
 IS_ROCM = torch.version.hip is not None
+IS_MPS = torch.backends.mps.is_available()
 
 
 class Platform(ABC):
@@ -32,6 +33,9 @@ class Platform(ABC):
     @abstractmethod
     def is_rocm(self) -> bool: ...
 
+    @abstractmethod
+    def is_mps(self) -> bool: ...
+
 
 class CudaPlatform(Platform):
     @classmethod
@@ -45,6 +49,9 @@ class CudaPlatform(Platform):
     def is_rocm(self) -> bool:
         return False
 
+    def is_mps(self) -> bool:
+        return False
+
 
 class RocmPlatform(Platform):
     @classmethod
@@ -58,5 +65,28 @@ class RocmPlatform(Platform):
     def is_rocm(self) -> bool:
         return True
 
+    def is_mps(self) -> bool:
+        return False
+
+
+class MpsPlatform(Platform):
+    @classmethod
+    @lru_cache(maxsize=8)
+    def get_device_name(cls, device_id: int = 0) -> str:
+        return torch.cuda.get_device_name(device_id)
+
+    def is_cuda(self) -> bool:
+        return False
+
+    def is_rocm(self) -> bool:
+        return False
+
+    def is_mps(self) -> bool:
+        return True
 
-current_platform = RocmPlatform() if IS_ROCM else CudaPlatform()
+current_platform = (
+    RocmPlatform() if IS_ROCM else
+    MpsPlatform() if IS_MPS else
+    CudaPlatform() if torch.cuda.is_available() else
+    None
+)

build/torch27-cxx11-cu126-aarch64-linux/paged_attention/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _paged_attention_daf6221
-ops = torch.ops._paged_attention_daf6221
+from . import _paged_attention_6677800
+ops = torch.ops._paged_attention_6677800
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_paged_attention_daf6221::{op_name}"
+    return f"_paged_attention_6677800::{op_name}"

build/torch27-cxx11-cu126-aarch64-linux/paged_attention/{_paged_attention_daf6221.abi3.so → _paged_attention_6677800.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:71c360e1d9268a0794c51393fd1d943472952abb5748af37a243a3f06155fb8c
-size 87576232
+oid sha256:2639286d960d9992c40b85c911773825374d8bb218a82eb84cdd4ac41342cb4c
+size 88169904

build/torch27-cxx11-cu126-aarch64-linux/paged_attention/platforms.py

@@ -8,6 +8,7 @@ import numpy as np
 import torch
 
 IS_ROCM = torch.version.hip is not None
+IS_MPS = torch.backends.mps.is_available()
 
 
 class Platform(ABC):
@@ -32,6 +33,9 @@ class Platform(ABC):
     @abstractmethod
     def is_rocm(self) -> bool: ...
 
+    @abstractmethod
+    def is_mps(self) -> bool: ...
+
 
 class CudaPlatform(Platform):
     @classmethod
@@ -45,6 +49,9 @@ class CudaPlatform(Platform):
     def is_rocm(self) -> bool:
         return False
 
+    def is_mps(self) -> bool:
+        return False
+
 
 class RocmPlatform(Platform):
     @classmethod
@@ -58,5 +65,28 @@ class RocmPlatform(Platform):
     def is_rocm(self) -> bool:
         return True
 
+    def is_mps(self) -> bool:
+        return False
+
+
+class MpsPlatform(Platform):
+    @classmethod
+    @lru_cache(maxsize=8)
+    def get_device_name(cls, device_id: int = 0) -> str:
+        return torch.cuda.get_device_name(device_id)
+
+    def is_cuda(self) -> bool:
+        return False
+
+    def is_rocm(self) -> bool:
+        return False
+
+    def is_mps(self) -> bool:
+        return True
 
-current_platform = RocmPlatform() if IS_ROCM else CudaPlatform()
+current_platform = (
+    RocmPlatform() if IS_ROCM else
+    MpsPlatform() if IS_MPS else
+    CudaPlatform() if torch.cuda.is_available() else
+    None
+)

build/torch27-cxx11-cu128-aarch64-linux/paged_attention/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _paged_attention_daf6221
-ops = torch.ops._paged_attention_daf6221
+from . import _paged_attention_6677800
+ops = torch.ops._paged_attention_6677800
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_paged_attention_daf6221::{op_name}"
+    return f"_paged_attention_6677800::{op_name}"

build/torch27-cxx11-cu128-aarch64-linux/paged_attention/{_paged_attention_daf6221.abi3.so → _paged_attention_6677800.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d39b07d716e2499580fdf6edf14af97abaef398453e77fd7d360e6d6bcc12df1
-size 119164016
+oid sha256:7bb98e687cc11950f2103990b777e6699c4a7285a53092def0063a9684fa951e
+size 120216032

build/torch27-cxx11-cu128-aarch64-linux/paged_attention/platforms.py

@@ -8,6 +8,7 @@ import numpy as np
 import torch
 
 IS_ROCM = torch.version.hip is not None
+IS_MPS = torch.backends.mps.is_available()
 
 
 class Platform(ABC):
@@ -32,6 +33,9 @@ class Platform(ABC):
     @abstractmethod
     def is_rocm(self) -> bool: ...
 
+    @abstractmethod
+    def is_mps(self) -> bool: ...
+
 
 class CudaPlatform(Platform):
     @classmethod
@@ -45,6 +49,9 @@ class CudaPlatform(Platform):
     def is_rocm(self) -> bool:
         return False
 
+    def is_mps(self) -> bool:
+        return False
+
 
 class RocmPlatform(Platform):
     @classmethod
@@ -58,5 +65,28 @@ class RocmPlatform(Platform):
     def is_rocm(self) -> bool:
         return True
 
+    def is_mps(self) -> bool:
+        return False
+
+
+class MpsPlatform(Platform):
+    @classmethod
+    @lru_cache(maxsize=8)
+    def get_device_name(cls, device_id: int = 0) -> str:
+        return torch.cuda.get_device_name(device_id)
+
+    def is_cuda(self) -> bool:
+        return False
+
+    def is_rocm(self) -> bool:
+        return False
+
+    def is_mps(self) -> bool:
+        return True
 
-current_platform = RocmPlatform() if IS_ROCM else CudaPlatform()
+current_platform = (
+    RocmPlatform() if IS_ROCM else
+    MpsPlatform() if IS_MPS else
+    CudaPlatform() if torch.cuda.is_available() else
+    None
+)