logging ops

Author: eightysteele

.github/workflows/ci.yml

@@ -74,9 +74,9 @@ jobs:
           source ~/.bashrc
           export PATH="/home/runner/.pixi/bin:/home/runner/.local/bin:$PATH"
           export CPLUS_INCLUDE_PATH="$GITHUB_WORKSPACE/.pixi/envs/default/targets/x86_64-linux/include"
-          sudo cp -v "$GITHUB_WORKSPACE/assets/system/10_nvidia.json" /usr/share/glvnd/egl_vendor.d/
+          #sudo cp -v "$GITHUB_WORKSPACE/assets/system/10_nvidia.json" /usr/share/glvnd/egl_vendor.d/
           pixi tree
-          export LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu:$LD_LIBRARY_PATH
+          export LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu
           env
           pixi run -vv test-ci
 

pixi.lock

@@ -69,7 +69,7 @@ b3d-pull = {cmd = "python b3d_pull.py -ow", cwd = "src/b3d/bucket_utils" }
 
 rerun = "rerun --port 8812"
 
-test = { cmd = "pytest -s --log-cli-level=DEBUG tests/dense_model_unit_tests/triangle_depth_posterior/test_triangle_depth_posterior.py", env = { XLA_PYTHON_CLIENT_PREALLOCATE = "false", XLA_PYTHON_CLIENT_ALLOCATOR = "platform", CPLUS_INCLUDE_PATH = "./targets/x86_64-linux/include", TORCH_CUDA_ARCH_LIST = "8.9", LD_LIBRARY_PATH = "/usr/lib/x86_64-linux-gnu:$LD_LIBRARY_PATH" } }
+test = { cmd = "pytest -s --log-cli-level=WARN tests/dense_model_unit_tests/triangle_depth_posterior/test_triangle_depth_posterior.py", env = { XLA_PYTHON_CLIENT_PREALLOCATE = "false", XLA_PYTHON_CLIENT_ALLOCATOR = "platform", CPLUS_INCLUDE_PATH = "./targets/x86_64-linux/include", TORCH_CUDA_ARCH_LIST = "8.9", LD_LIBRARY_PATH = "/usr/lib/x86_64-linux-gnu" } }
 
 test-ci = { cmd = "pytest -s --log-cli-level=DEBUG tests/dense_model_unit_tests/triangle_depth_posterior/test_triangle_depth_posterior.py", env = { XLA_PYTHON_CLIENT_PREALLOCATE = "false", XLA_PYTHON_CLIENT_ALLOCATOR = "platform", TORCH_CUDA_ARCH_LIST = "8.9"} }
 

pyproject.toml

@@ -213,6 +213,13 @@ def __init__(self, output_db=True, mode="automatic", device=None):
         else:
             with torch.cuda.device(device):
                 cuda_device_idx = torch.cuda.current_device()
+
+        plugin = _get_plugin(gl=True)
+
+        logging.getLogger("nvdiffrast").warning(
+            f"cuda_device_idx: {cuda_device_idx}, plugin: {plugin}, gl: {plugin.RasterizeGLStateWrapper}"
+        )
+
         self.cpp_wrapper = _get_plugin(gl=True).RasterizeGLStateWrapper(
             output_db, mode == "automatic", cuda_device_idx
         )