put python in path

EnzymeAD · Oct 15, 2024 · 2042008 · 2042008
1 parent cb6156d
commit 2042008
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.buildkite/gpu_pipeline.yml b/.buildkite/gpu_pipeline.yml
@@ -53,6 +53,7 @@ steps:
       export LD_LIBRARY_PATH="`pwd`/bazel-bin/test/llama.runfiles/pypi_nvidia_cuda_runtime_cu12/site-packages/nvidia/cuda_runtime/lib:\$LD_LIBRARY_PATH"
       export LD_LIBRARY_PATH="`pwd`/bazel-bin/test/llama.runfiles/pypi_nvidia_cuda_nvrtc_cu12/site-packages/nvidia/cuda_nvrtc/lib:\$LD_LIBRARY_PATH"
       export PATH="`pwd`/bazel-bin/test/llama.runfiles/pypi_nvidia_cuda_nvcc_cu12/site-packages/nvidia/cuda_nvcc/bin:\$PATH"
+      export PATH="`pwd`/bazel-bin/test/llama.runfiles/python_x86_64-unknown-linux-gnu/bin:\$PATH"
       export TF_CPP_MIN_LOG_LEVEL=0
 
       BAZEL_BUILD_FLAGS=()
@@ -91,3 +92,4 @@ steps:
       - "finds.txt"
       - "bazel-out/*/testlogs/test/llama/test.log"
       - "bazel-out/*/testlogs/test/llama/bench_vs_xla.log"
+      - "results_*.csv"
diff --git a/src/enzyme_ad/jax/deps/tensat/src/input.rs b/src/enzyme_ad/jax/deps/tensat/src/input.rs
@@ -1325,7 +1325,7 @@ fn extract_by_ilp(
     arg_vec.push(time_lim);
     arg_vec.push("--num_thread");
     arg_vec.push(num_thread);
-    let child = Command::new("python")
+    let child = Command::new("python3")
         .args(&arg_vec)
         .spawn()
         .expect("failed to execute child");