Mshugrina/minor fixes (#814)

* Minor docs and bug fixes pre-release

Signed-off-by: Maria Masha Shugrina <[email protected]>

* add little adjustement on doc

Signed-off-by: Clement Fuji Tsang <[email protected]>

---------

Signed-off-by: Maria Masha Shugrina <[email protected]>
Signed-off-by: Clement Fuji Tsang <[email protected]>
Co-authored-by: Maria Masha Shugrina <[email protected]>

Author: Caenorst

docs/kaolin_ext.py

@@ -75,8 +75,11 @@ def run_apidoc(_):
             "kaolin/physics/utils/scene_forces.py",
             "kaolin/render/lighting/sg.py",
             "kaolin/render/lighting/sh.py",
+            "kaolin/render/easy_render/common.py",
+            "kaolin/render/easy_render/mesh.py",
             "kaolin/render/mesh/deftet.py",
             "kaolin/render/mesh/dibr.py",
+            "kaolin/render/mesh/nvdiffrast_context.py",
             "kaolin/render/mesh/rasterization.py",
             "kaolin/render/mesh/utils.py",
             "kaolin/render/spc/raytrace.py",

docs/modules/kaolin.ops.rst

@@ -3,7 +3,7 @@
 kaolin.ops
 ==========
 
-Operators are primitive processing functions for batched 3D models (:ref:`meshes<kaolin.ops.mesh>`, :ref:`voxelgrids<kaolin.ops.voxelgrid>` and point clouds).
+Operators are primitive processing functions for batched 3D models (:ref:`meshes<kaolin.ops.mesh>`, ::ref:`voxelgrids<kaolin.ops.voxelgrid>` and point clouds).
 
 Tensor batching operators are in :ref:`kaolin.ops.batch`, conversions of 3D models between different representations are in :ref:`kaolin.ops.conversions`.
 

docs/modules/kaolin.render.easy_render.rst

@@ -0,0 +1,29 @@
+kaolin.render.easy\_render
+==========================
+
+Common Utilities
+----------------
+
+.. autofunction:: kaolin.render.easy_render.default_camera
+.. autofunction:: kaolin.render.easy_render.default_lighting
+.. autofunction:: kaolin.render.easy_render.default_material
+
+
+Easy Rendering of Meshes
+------------------------
+
+.. autoclass:: kaolin.render.easy_render.RenderPass
+   :members:
+   :undoc-members:
+
+.. autofunction:: kaolin.render.easy_render.render_mesh
+
+
+Misc Utilities
+--------------
+.. automodule:: kaolin.render.easy_render
+   :members:
+   :member-order: bysource
+   :undoc-members:
+   :show-inheritance:
+   :exclude-members: default_camera, default_lighting, default_material, render_mesh, RenderPass

kaolin/render/easy_render/__init__.py

@@ -1,2 +1,4 @@
 from .common import *
 from .mesh import *
+
+__all__ = [k for k in locals().keys() if not k.startswith('__')]

kaolin/render/easy_render/common.py

@@ -20,6 +20,8 @@
 from kaolin.render.lighting import SgLightingParameters, sg_direction_from_azimuth_elevation
 from kaolin.render.materials import PBRMaterial
 
+__all__ = ['default_lighting', 'default_camera', 'default_material']
+
 
 def default_lighting():
     """ Returns default lighting, represented as Spherical Gaussians.

kaolin/render/easy_render/mesh.py

@@ -28,6 +28,10 @@
     import nvdiffrast.torch
 
 
+__all__ = ['render_mesh', 'mesh_rasterize_interpolate_cuda', 'mesh_rasterize_interpolate_nvdiffrast',
+           'texture_sample_materials', 'sg_shade', 'RenderPass']
+
+
 class RenderPass(str, Enum):
     # TODO: add normals relative to camera, mask, anti-aliasing, depth, wireframe
     face_idx = "face_idx"

kaolin/render/mesh/nvdiffrast_context.py

@@ -42,6 +42,7 @@ def nvdiffrast_is_available():
 
 def nvdiffrast_use_cuda():
     """ Configures nvdiffrast back end to use `nvdiffrast.torch.RasterizeCudaContext` by default."""
+    global _default_context_fn
     if nvdiffrast_is_available():
         _default_context_fn = nvdiff.RasterizeCudaContext
     else:
@@ -50,6 +51,7 @@ def nvdiffrast_use_cuda():
 
 def nvdiffrast_use_opengl():
     """ Configures nvdiffrast back end to use `nvdiffrast.torch.RasterizeGLContext` by default."""
+    global _default_context_fn
     if nvdiffrast_is_available():
         _default_context_fn = partial(nvdiff.RasterizeGLContext, output_db=False)
     else:

tools/linux/run_tests.sh

@@ -166,8 +166,8 @@ if [ $BUILD_DOCS -eq "1" ]; then
     CMDLINE="python -m sphinx -T -E -W --keep-going -b html -d _build/doctrees -D language=en . _build/html"
     export PYTORCH_JIT=0
     $CMDLINE >> $DOCS_LOG 2>&1
-    export PYTORCH_JIT=1
     RES=$?
+    export PYTORCH_JIT=1
 
     cd $KAOLIN_ROOT
     DOCS_URL="build_docs/_build/html/index.html"