From 2b144118c78617bbf181a6f7d6d0f4499380c25d Mon Sep 17 00:00:00 2001
From: Aaron Steele <eightysteele@gmail.com>
Date: Wed, 17 Jul 2024 06:57:21 -0700
Subject: [PATCH] check before ci

---
 pixi.lock                 | 2 +-
 pyproject.toml            | 5 ++---
 scripts/env-activation.sh | 4 ++++
 3 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/pixi.lock b/pixi.lock
index 197df8c3..431161e9 100644
--- a/pixi.lock
+++ b/pixi.lock
@@ -1156,7 +1156,7 @@ packages:
   name: b3d-prototype
   version: 0.1.0
   path: .
-  sha256: bdd715a11b829d0cd12a913edbd8290fda18c4eea572252a04c421d77fd19409
+  sha256: c33c185c67b89b15e3c413683764562b67c22b38e7bc41ca941ddb5805379847
   requires_dist:
   - genjax==0.5.0.post13.dev0+973fb60d
   - opencv-python>=4.10.0.84,<4.10.1
diff --git a/pyproject.toml b/pyproject.toml
index 2be9c549..1e6ece03 100644
--- 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"]
\ No newline at end of file
diff --git a/scripts/env-activation.sh b/scripts/env-activation.sh
index 408dc599..9924cd02 100755
--- 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"