diff --git a/docs/conf.py b/docs/conf.py
index 1c1f8afc..c06e6516 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -12,7 +12,7 @@
 
 sys.path.insert(0, os.path.abspath("../python"))
 os.environ["BUILD_DOC"] = "1"
-autodoc_mock_imports = ["torch"]
+autodoc_mock_imports = ["torch", "triton"]
 
 project = "FlashInfer"
 author = "FlashInfer Contributors"