diff --git a/core/runtime/py_attention_metadata.cpp b/core/runtime/py_attention_metadata.cpp
index 335e89e3..259cdae6 100644
--- a/core/runtime/py_attention_metadata.cpp
+++ b/core/runtime/py_attention_metadata.cpp
@@ -341,3 +341,8 @@ py::object PyAttentionMetadataView::optional_tensor(
 }
 
 }  // namespace project6
+
+PYBIND11_MODULE(py_attention_metadata, m) {
+  m.doc() = "DP-aware attention metadata (project6, ported from xLLM PR #2258)";
+  project6::register_attention_metadata_views(m);
+}