Update active_rotated_filter_cuda_kernel.cuh

dflhw · Jun 24, 2024 · d609aa8 · d609aa8
1 parent 8f23a0b
commit d609aa8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mmcv/ops/csrc/common/cuda/active_rotated_filter_cuda_kernel.cuh b/mmcv/ops/csrc/common/cuda/active_rotated_filter_cuda_kernel.cuh
@@ -21,6 +21,7 @@ __global__ void active_rotated_filter_forward_cuda_kernel(
     int j = (index / nEntry) % num_input_planes;
     int i = index / nEntry / num_input_planes;
     int k;
+    int flag = 1;
     scalar_t val = *(weight_data + index);
     for (k = 0; k < num_rotations; k++) {
       int idx = (int)(*(indices_data + l * num_rotations + k)) - 1;