bulatko commited on
Commit
f56c6ac
·
1 Parent(s): 644503f

refactor(cuda): revert verbose assertion comments

Browse files
MaskClustering/third_party/detectron2/projects/CropFormer/mask2former/modeling/pixel_decoder/ops/src/cuda/ms_deform_attn_cuda.cu CHANGED
@@ -36,7 +36,7 @@ at::Tensor ms_deform_attn_cuda_forward(
36
  AT_ASSERTM(sampling_loc.is_contiguous(), "sampling_loc tensor has to be contiguous");
37
  AT_ASSERTM(attn_weight.is_contiguous(), "attn_weight tensor has to be contiguous");
38
 
39
- AT_ASSERTM(value.is_cuda() // PyTorch 2.8 compat, "value must be a CUDA tensor");
40
  AT_ASSERTM(spatial_shapes.is_cuda(), "spatial_shapes must be a CUDA tensor");
41
  AT_ASSERTM(level_start_index.is_cuda(), "level_start_index must be a CUDA tensor");
42
  AT_ASSERTM(sampling_loc.is_cuda(), "sampling_loc must be a CUDA tensor");
@@ -102,7 +102,7 @@ std::vector<at::Tensor> ms_deform_attn_cuda_backward(
102
  AT_ASSERTM(attn_weight.is_contiguous(), "attn_weight tensor has to be contiguous");
103
  AT_ASSERTM(grad_output.is_contiguous(), "grad_output tensor has to be contiguous");
104
 
105
- AT_ASSERTM(value.is_cuda() // PyTorch 2.8 compat, "value must be a CUDA tensor");
106
  AT_ASSERTM(spatial_shapes.is_cuda(), "spatial_shapes must be a CUDA tensor");
107
  AT_ASSERTM(level_start_index.is_cuda(), "level_start_index must be a CUDA tensor");
108
  AT_ASSERTM(sampling_loc.is_cuda(), "sampling_loc must be a CUDA tensor");
 
36
  AT_ASSERTM(sampling_loc.is_contiguous(), "sampling_loc tensor has to be contiguous");
37
  AT_ASSERTM(attn_weight.is_contiguous(), "attn_weight tensor has to be contiguous");
38
 
39
+ AT_ASSERTM(value.is_cuda(), "value must be a CUDA tensor");
40
  AT_ASSERTM(spatial_shapes.is_cuda(), "spatial_shapes must be a CUDA tensor");
41
  AT_ASSERTM(level_start_index.is_cuda(), "level_start_index must be a CUDA tensor");
42
  AT_ASSERTM(sampling_loc.is_cuda(), "sampling_loc must be a CUDA tensor");
 
102
  AT_ASSERTM(attn_weight.is_contiguous(), "attn_weight tensor has to be contiguous");
103
  AT_ASSERTM(grad_output.is_contiguous(), "grad_output tensor has to be contiguous");
104
 
105
+ AT_ASSERTM(value.is_cuda(), "value must be a CUDA tensor");
106
  AT_ASSERTM(spatial_shapes.is_cuda(), "spatial_shapes must be a CUDA tensor");
107
  AT_ASSERTM(level_start_index.is_cuda(), "level_start_index must be a CUDA tensor");
108
  AT_ASSERTM(sampling_loc.is_cuda(), "sampling_loc must be a CUDA tensor");