check before ci

probcomp · Jul 17, 2024 · 2b14411 · 2b14411
1 parent 3b5e1ea
commit 2b14411
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/pixi.lock b/pixi.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -63,10 +63,9 @@ ninja = "*"
 pytest = "*"
 
 [tool.pixi.tasks]
-foo = "export FOO=8"
-rerun = "rerun --port 8812"
 b3d-pull = {cmd = "python b3d_pull.py -ow", cwd = "b3d/bucket_utils" }
-test = { cmd = "pytest 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 = "$CONDA_PREFIX/targets/x86_64-linux/include"} }
+rerun = "rerun --port 8812"
+test = { cmd = "pytest 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 = "$CONDA_PREFIX/targets/x86_64-linux/include", TORCH_CUDA_ARCH_LIST = "8.9"} }
 
 [tool.pytest.ini_options]
 pythonpath = ["src"]
diff --git a/scripts/env-activation.sh b/scripts/env-activation.sh
@@ -5,3 +5,7 @@ set -euo pipefail
 capability=$(nvidia-smi --query-gpu=compute_cap --format=csv,noheader)
 
 export TORCH_CUDA_ARCH_LIST="$capability"
+export XLA_FLAGS=--xla_gpu_enable_command_buffer=
+export XLA_PYTHON_CLIENT_PREALLOCATE=false
+export XLA_PYTHON_CLIENT_ALLOCATOR=platform
+export CPLUS_INCLUDE_PATH="$CONDA_PREFIX/targets/x86_64-linux/include"