diff --git a/torch/CMakeLists.txt b/torch/CMakeLists.txt
index 8b8ebdc6e97..67fe1df8ca8 100644
--- a/torch/CMakeLists.txt
+++ b/torch/CMakeLists.txt
@@ -251,6 +251,8 @@ add_custom_command(
       "${TORCH_ROOT}/aten/src/ATen/native/native_functions.yaml"
       "${TORCH_ROOT}/aten/src/ATen/native/tags.yaml"
       "${TORCH_ROOT}/tools/autograd/deprecated.yaml"
+      "${TORCH_ROOT}/torch/_torch_docs.py"
+      "${TORCH_ROOT}/torch/_tensor_docs.py"
       ${pyi_python}
       ${autograd_python}
       ${torchgen_python}