Build uploaded using `kernels`.

build/torch27-cxx11-cu118-x86_64-linux/deformable_detr/{_deformable_detr_7c33cbe.abi3.so → _deformable_detr_57c3d32.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:20ee462e214e11abd65b44259847f7dbf535b94a3003357251c53a72f4ac4392
-size 6693728
+oid sha256:9293f97cc6b06bc3ba5e57cfd084abb252c287f4518935208e67e126e7cbd19b
+size 6800224

build/torch27-cxx11-cu118-x86_64-linux/deformable_detr/_ops.py

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

build/torch27-cxx11-cu126-x86_64-linux/deformable_detr/{_deformable_detr_7c33cbe.abi3.so → _deformable_detr_57c3d32.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c2a2064d5d7b5f62704d4cad6a530dd812804cfcc426889030fefdcc553828d3
-size 6800688
+oid sha256:df54f46f59b5b78b15314cb0825d8f1f34c7a3198e9d62ca2a65a8ca72ea79a4
+size 6911280

build/torch27-cxx11-cu126-x86_64-linux/deformable_detr/_ops.py

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

build/torch27-cxx11-cu128-x86_64-linux/deformable_detr/{_deformable_detr_7c33cbe.abi3.so → _deformable_detr_57c3d32.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1f819d4d59e9ea2ce0afd8a0198784cb9f1c3100a0d352261bf84d5b8570cda7
-size 9796776
+oid sha256:6f389536870cd4acf36ab8f12d3b0bf9f847ec06e2cfc25905420796884b614e
+size 9907368

build/torch27-cxx11-cu128-x86_64-linux/deformable_detr/_ops.py

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

build/torch28-cxx11-cu126-x86_64-linux/deformable_detr/{_deformable_detr_e82640e_dirty.abi3.so → _deformable_detr_57c3d32.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8877a2159e6f4f04905120af8d10ab9b8294aff80b97a584d6a6716922141b46
-size 6802152
+oid sha256:849b408dcea47846a61de3e2c5dc178297e38ac85542e99d737887da93b199e6
+size 6920920

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

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

build/torch28-cxx11-cu128-x86_64-linux/deformable_detr/_deformable_detr_57c3d32.abi3.so

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

build/torch28-cxx11-cu128-x86_64-linux/deformable_detr/_deformable_detr_e82640e_dirty.abi3.so

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

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

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

build/torch28-cxx11-cu129-x86_64-linux/deformable_detr/_deformable_detr_57c3d32.abi3.so

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

build/torch28-cxx11-cu129-x86_64-linux/deformable_detr/_deformable_detr_e82640e_dirty.abi3.so

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

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

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

build/torch29-cxx11-cu126-x86_64-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/torch29-cxx11-cu126-x86_64-linux/deformable_detr/_deformable_detr_57c3d32.abi3.so

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

build/torch29-cxx11-cu126-x86_64-linux/deformable_detr/_ops.py

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

build/torch29-cxx11-cu126-x86_64-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/torch29-cxx11-cu128-x86_64-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/torch29-cxx11-cu128-x86_64-linux/deformable_detr/_deformable_detr_57c3d32.abi3.so

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

build/torch29-cxx11-cu128-x86_64-linux/deformable_detr/_ops.py

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

build/torch29-cxx11-cu128-x86_64-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/torch29-cxx11-cu130-x86_64-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"]