Build (aarch64-linux)

build/torch28-cxx11-cu126-aarch64-linux/deformable_detr/__init__.py

@@ -0,0 +1,46 @@
+from typing import List
+import torch
+
+from ._ops import ops
+from . import layers
+
+
+def ms_deform_attn_backward(
+    value: torch.Tensor,
+    spatial_shapes: torch.Tensor,
+    level_start_index: torch.Tensor,
+    sampling_loc: torch.Tensor,
+    attn_weight: torch.Tensor,
+    grad_output: torch.Tensor,
+    im2col_step: int,
+) -> List[torch.Tensor]:
+    return ops.ms_deform_attn_backward(
+        value,
+        spatial_shapes,
+        level_start_index,
+        sampling_loc,
+        attn_weight,
+        grad_output,
+        im2col_step,
+    )
+
+
+def ms_deform_attn_forward(
+    value: torch.Tensor,
+    spatial_shapes: torch.Tensor,
+    level_start_index: torch.Tensor,
+    sampling_loc: torch.Tensor,
+    attn_weight: torch.Tensor,
+    im2col_step: int,
+) -> torch.Tensor:
+    return ops.ms_deform_attn_forward(
+        value,
+        spatial_shapes,
+        level_start_index,
+        sampling_loc,
+        attn_weight,
+        im2col_step,
+    )
+
+
+__all__ = ["layers", "ms_deform_attn_forward", "ms_deform_attn_backward"]

build/torch28-cxx11-cu126-aarch64-linux/deformable_detr/_deformable_detr_a92c8ea_dirty.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b4a5665b05309312200ca97a80cc61340c0f5de123ab33254e5307a5ec4ed2a0
+size 6901024

build/torch28-cxx11-cu126-aarch64-linux/deformable_detr/_ops.py

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

build/torch28-cxx11-cu126-aarch64-linux/deformable_detr/layers.py

@@ -0,0 +1,84 @@
+from typing import List, Union, Tuple
+
+from torch import Tensor
+from torch.autograd import Function
+from torch.autograd.function import once_differentiable
+import torch.nn as nn
+
+from ._ops import ops
+
+
+class MultiScaleDeformableAttentionFunction(Function):
+    @staticmethod
+    def forward(
+        context,
+        value: Tensor,
+        value_spatial_shapes: Tensor,
+        value_level_start_index: Tensor,
+        sampling_locations: Tensor,
+        attention_weights: Tensor,
+        im2col_step: int,
+    ):
+        context.im2col_step = im2col_step
+        output = ops.ms_deform_attn_forward(
+            value,
+            value_spatial_shapes,
+            value_level_start_index,
+            sampling_locations,
+            attention_weights,
+            context.im2col_step,
+        )
+        context.save_for_backward(
+            value,
+            value_spatial_shapes,
+            value_level_start_index,
+            sampling_locations,
+            attention_weights,
+        )
+        return output
+
+    @staticmethod
+    @once_differentiable
+    def backward(context, grad_output):
+        (
+            value,
+            value_spatial_shapes,
+            value_level_start_index,
+            sampling_locations,
+            attention_weights,
+        ) = context.saved_tensors
+        grad_value, grad_sampling_loc, grad_attn_weight = ops.ms_deform_attn_backward(
+            value,
+            value_spatial_shapes,
+            value_level_start_index,
+            sampling_locations,
+            attention_weights,
+            grad_output,
+            context.im2col_step,
+        )
+
+        return grad_value, None, None, grad_sampling_loc, grad_attn_weight, None
+
+
+class MultiScaleDeformableAttention(nn.Module):
+    def forward(
+        self,
+        value: Tensor,
+        value_spatial_shapes: Tensor,
+        value_spatial_shapes_list: List[Tuple],
+        level_start_index: Tensor,
+        sampling_locations: Tensor,
+        attention_weights: Tensor,
+        im2col_step: int,
+    ):
+        return MultiScaleDeformableAttentionFunction.apply(
+            value,
+            value_spatial_shapes,
+            level_start_index,
+            sampling_locations,
+            attention_weights,
+            im2col_step,
+        )
+
+
+__all__ = ["MultiScaleDeformableAttention"]

build/torch28-cxx11-cu128-aarch64-linux/deformable_detr/__init__.py

@@ -0,0 +1,46 @@
+from typing import List
+import torch
+
+from ._ops import ops
+from . import layers
+
+
+def ms_deform_attn_backward(
+    value: torch.Tensor,
+    spatial_shapes: torch.Tensor,
+    level_start_index: torch.Tensor,
+    sampling_loc: torch.Tensor,
+    attn_weight: torch.Tensor,
+    grad_output: torch.Tensor,
+    im2col_step: int,
+) -> List[torch.Tensor]:
+    return ops.ms_deform_attn_backward(
+        value,
+        spatial_shapes,
+        level_start_index,
+        sampling_loc,
+        attn_weight,
+        grad_output,
+        im2col_step,
+    )
+
+
+def ms_deform_attn_forward(
+    value: torch.Tensor,
+    spatial_shapes: torch.Tensor,
+    level_start_index: torch.Tensor,
+    sampling_loc: torch.Tensor,
+    attn_weight: torch.Tensor,
+    im2col_step: int,
+) -> torch.Tensor:
+    return ops.ms_deform_attn_forward(
+        value,
+        spatial_shapes,
+        level_start_index,
+        sampling_loc,
+        attn_weight,
+        im2col_step,
+    )
+
+
+__all__ = ["layers", "ms_deform_attn_forward", "ms_deform_attn_backward"]

build/torch28-cxx11-cu128-aarch64-linux/deformable_detr/_deformable_detr_a92c8ea_dirty.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3f4a0a7850968822e26e3a59c801fd711231d5294193155efbf9583761e114ef
+size 9849688

build/torch28-cxx11-cu128-aarch64-linux/deformable_detr/_ops.py

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

build/torch28-cxx11-cu128-aarch64-linux/deformable_detr/layers.py

@@ -0,0 +1,84 @@
+from typing import List, Union, Tuple
+
+from torch import Tensor
+from torch.autograd import Function
+from torch.autograd.function import once_differentiable
+import torch.nn as nn
+
+from ._ops import ops
+
+
+class MultiScaleDeformableAttentionFunction(Function):
+    @staticmethod
+    def forward(
+        context,
+        value: Tensor,
+        value_spatial_shapes: Tensor,
+        value_level_start_index: Tensor,
+        sampling_locations: Tensor,
+        attention_weights: Tensor,
+        im2col_step: int,
+    ):
+        context.im2col_step = im2col_step
+        output = ops.ms_deform_attn_forward(
+            value,
+            value_spatial_shapes,
+            value_level_start_index,
+            sampling_locations,
+            attention_weights,
+            context.im2col_step,
+        )
+        context.save_for_backward(
+            value,
+            value_spatial_shapes,
+            value_level_start_index,
+            sampling_locations,
+            attention_weights,
+        )
+        return output
+
+    @staticmethod
+    @once_differentiable
+    def backward(context, grad_output):
+        (
+            value,
+            value_spatial_shapes,
+            value_level_start_index,
+            sampling_locations,
+            attention_weights,
+        ) = context.saved_tensors
+        grad_value, grad_sampling_loc, grad_attn_weight = ops.ms_deform_attn_backward(
+            value,
+            value_spatial_shapes,
+            value_level_start_index,
+            sampling_locations,
+            attention_weights,
+            grad_output,
+            context.im2col_step,
+        )
+
+        return grad_value, None, None, grad_sampling_loc, grad_attn_weight, None
+
+
+class MultiScaleDeformableAttention(nn.Module):
+    def forward(
+        self,
+        value: Tensor,
+        value_spatial_shapes: Tensor,
+        value_spatial_shapes_list: List[Tuple],
+        level_start_index: Tensor,
+        sampling_locations: Tensor,
+        attention_weights: Tensor,
+        im2col_step: int,
+    ):
+        return MultiScaleDeformableAttentionFunction.apply(
+            value,
+            value_spatial_shapes,
+            level_start_index,
+            sampling_locations,
+            attention_weights,
+            im2col_step,
+        )
+
+
+__all__ = ["MultiScaleDeformableAttention"]

build/torch28-cxx11-cu129-aarch64-linux/deformable_detr/__init__.py

@@ -0,0 +1,46 @@
+from typing import List
+import torch
+
+from ._ops import ops
+from . import layers
+
+
+def ms_deform_attn_backward(
+    value: torch.Tensor,
+    spatial_shapes: torch.Tensor,
+    level_start_index: torch.Tensor,
+    sampling_loc: torch.Tensor,
+    attn_weight: torch.Tensor,
+    grad_output: torch.Tensor,
+    im2col_step: int,
+) -> List[torch.Tensor]:
+    return ops.ms_deform_attn_backward(
+        value,
+        spatial_shapes,
+        level_start_index,
+        sampling_loc,
+        attn_weight,
+        grad_output,
+        im2col_step,
+    )
+
+
+def ms_deform_attn_forward(
+    value: torch.Tensor,
+    spatial_shapes: torch.Tensor,
+    level_start_index: torch.Tensor,
+    sampling_loc: torch.Tensor,
+    attn_weight: torch.Tensor,
+    im2col_step: int,
+) -> torch.Tensor:
+    return ops.ms_deform_attn_forward(
+        value,
+        spatial_shapes,
+        level_start_index,
+        sampling_loc,
+        attn_weight,
+        im2col_step,
+    )
+
+
+__all__ = ["layers", "ms_deform_attn_forward", "ms_deform_attn_backward"]

build/torch28-cxx11-cu129-aarch64-linux/deformable_detr/_deformable_detr_a92c8ea_dirty.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bc660849ac99525cfc072e758b85d07ed8b3040e4e529efeb42dcd0d286690bd
+size 9916656

build/torch28-cxx11-cu129-aarch64-linux/deformable_detr/_ops.py

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

build/torch28-cxx11-cu129-aarch64-linux/deformable_detr/layers.py

@@ -0,0 +1,84 @@
+from typing import List, Union, Tuple
+
+from torch import Tensor
+from torch.autograd import Function
+from torch.autograd.function import once_differentiable
+import torch.nn as nn
+
+from ._ops import ops
+
+
+class MultiScaleDeformableAttentionFunction(Function):
+    @staticmethod
+    def forward(
+        context,
+        value: Tensor,
+        value_spatial_shapes: Tensor,
+        value_level_start_index: Tensor,
+        sampling_locations: Tensor,
+        attention_weights: Tensor,
+        im2col_step: int,
+    ):
+        context.im2col_step = im2col_step
+        output = ops.ms_deform_attn_forward(
+            value,
+            value_spatial_shapes,
+            value_level_start_index,
+            sampling_locations,
+            attention_weights,
+            context.im2col_step,
+        )
+        context.save_for_backward(
+            value,
+            value_spatial_shapes,
+            value_level_start_index,
+            sampling_locations,
+            attention_weights,
+        )
+        return output
+
+    @staticmethod
+    @once_differentiable
+    def backward(context, grad_output):
+        (
+            value,
+            value_spatial_shapes,
+            value_level_start_index,
+            sampling_locations,
+            attention_weights,
+        ) = context.saved_tensors
+        grad_value, grad_sampling_loc, grad_attn_weight = ops.ms_deform_attn_backward(
+            value,
+            value_spatial_shapes,
+            value_level_start_index,
+            sampling_locations,
+            attention_weights,
+            grad_output,
+            context.im2col_step,
+        )
+
+        return grad_value, None, None, grad_sampling_loc, grad_attn_weight, None
+
+
+class MultiScaleDeformableAttention(nn.Module):
+    def forward(
+        self,
+        value: Tensor,
+        value_spatial_shapes: Tensor,
+        value_spatial_shapes_list: List[Tuple],
+        level_start_index: Tensor,
+        sampling_locations: Tensor,
+        attention_weights: Tensor,
+        im2col_step: int,
+    ):
+        return MultiScaleDeformableAttentionFunction.apply(
+            value,
+            value_spatial_shapes,
+            level_start_index,
+            sampling_locations,
+            attention_weights,
+            im2col_step,
+        )
+
+
+__all__ = ["MultiScaleDeformableAttention"]