[Feature] ms_opt

open-mmlab · Dec 26, 2022 · a6ff625 · a6ff625
1 parent b70f205
commit a6ff625
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmcv/ops/csrc/common/mlu/ms_deform_attn_mlu_kernel.mlu b/mmcv/ops/csrc/common/mlu/ms_deform_attn_mlu_kernel.mlu
@@ -42,7 +42,7 @@
  ****************************************************************************************/
 
 #define TWELVE_SPLIT 12
-#define ALIGN_NUM 64
+#define ALIGN_NUM 32
 #define ALIGN_NUM_FOR_REDUCE 32
 #define LEN_FLOAT sizeof(float)