diff --git a/csrc/rocm_ops.cpp b/csrc/rocm_ops.cpp
index 7f89db3c93..cf649815d9 100644
--- a/csrc/rocm_ops.cpp
+++ b/csrc/rocm_ops.cpp
@@ -27,6 +27,7 @@
 #include "hipbsolgemm.cuh"
 #include "mla.h"
 #include "moe_ck.h"
+#include "moe_cktile2stages.h"
 #include "moe_op.h"
 #include "moe_sorting.h"
 #include "norm.h"
@@ -84,6 +85,7 @@ PYBIND11_MODULE(TORCH_EXTENSION_NAME, m)
     POS_ENCODING_PYBIND;
     ATTENTION_PYBIND;
     MOE_CK_2STAGES_PYBIND;
+    MOE_CKTILE_2STAGES_PYBIND;
     QUANT_PYBIND;
     ATTENTION_ASM_PYBIND;
     ATTENTION_RAGGED_PYBIND;