From 89569a8984ad8f3dcf1e0824d5ad6d3a3a5421a8 Mon Sep 17 00:00:00 2001
From: Paul Zhang <paulzhan@meta.com>
Date: Tue, 9 Jan 2024 11:53:40 -0800
Subject: [PATCH] Expose ZCH modules in Documentation (#1613)

Summary: Pull Request resolved: https://github.com/pytorch/torchrec/pull/1613

Differential Revision: D52603324
---
 docs/source/torchrec.modules.rst | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/docs/source/torchrec.modules.rst b/docs/source/torchrec.modules.rst
index 95dcf35db..e2c9ee771 100644
--- a/docs/source/torchrec.modules.rst
+++ b/docs/source/torchrec.modules.rst
@@ -76,10 +76,18 @@ torchrec.modules.utils
    :undoc-members:
    :show-inheritance:
 
-Module contents
+torchrec.modules.mc_modules
 ---------------
 
-.. automodule:: torchrec.modules
+.. automodule:: torchrec.modules.mc_modules
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+torchrec.modules.mc_embedding_modules
+---------------
+
+.. automodule:: torchrec.modules.mc_embedding_modules
    :members:
    :undoc-members:
    :show-inheritance: