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()
|
| 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()
|
| 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");
|