Publish API docs for more modules.

Author: Diptorup Deb

docs/_templates/autoapi/index.rst

@@ -7,6 +7,8 @@ This page contains auto-generated API reference documentation [#f1]_.
    :maxdepth: 2
 
    numba_dpex/kernel_api/index
+   numba_dpex/experimental/decorators/index
+   numba_dpex/experimental/launcher/index
 
    {% for page in pages %}
    {% if page.top_level_object and page.display %}

docs/source/conf.py

@@ -113,19 +113,10 @@
 
 # -- Auto API configurations ---------------------------------------------------
 
-
-# def skip_util_classes(app, what, name, obj, skip, options):
-#     if what == "module" and "experimental" in name:
-#         if what == "module" and "decorators" not in name:
-#             skip = True
-#     return skip
-
-
-# def setup(sphinx):
-#     sphinx.connect("autoapi-skip-member", skip_util_classes)
-
-
-autoapi_dirs = ["../../numba_dpex/kernel_api"]
+autoapi_dirs = [
+    "../../numba_dpex/kernel_api",
+    "../../numba_dpex/experimental",
+]
 autoapi_type = "python"
 
 autoapi_template_dir = "_templates/autoapi"
@@ -159,3 +150,14 @@ def prepare_jinja_env(jinja_env) -> None:
 
 
 autoapi_prepare_jinja_env = prepare_jinja_env
+
+
+def skip_member(app, what, name, obj, skip, options):
+    # skip submodules
+    if what == "module":
+        skip = True
+    return skip
+
+
+def setup(sphinx):
+    sphinx.connect("autoapi-skip-member", skip_member)

docs/source/programming_model.rst

@@ -228,11 +228,7 @@ For example, consider
 
 .. code:: python
 
-    # Use usm_type = 'device' to get USM-device allocation (default),
-    #     usm_type = 'shared' to get USM-shared allocation,
-    #     usm_type = 'host'   to get USM-host allocation
-    def dpt.empty(..., device=None, usm_type=None, sycl_queue=None) -> dpctl.tensor.usm_ndarray:
-        ...
+    # TODO
 
 The keyword ``device`` is `mandated by the array API
 <https://data-apis.org/array-api/latest/design_topics/device_support.html#syntax-for-device-assignment>`__.