Fix paths

matter-labs · Aug 16, 2024 · baea64c · baea64c
1 parent 42a6732
commit baea64c
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 5 deletions.
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
@@ -13,6 +13,7 @@ jobs:
       image: nvidia/cuda:${{ matrix.cuda }}
     env:
       BELLMAN_CUDA_DIR: ${{ github.workspace }}/bellman-cuda
+      CUDAARCHS: 89
     steps:
       - name: Prepare environment
         env:

diff --git a/.github/workflows/test-gpu.yaml b/.github/workflows/test-gpu.yaml
@@ -97,6 +97,8 @@ jobs:
         run: |
           echo "/usr/local/nvidia/bin:/usr/local/cuda/bin" >> $GITHUB_PATH
 
+      - uses: actions/checkout@93ea575cb5d8a053eaa0ac8fa3b40d7e05a33cc8 # v3
+
       - name: Check Nvidia driver version
         run: |
           nvidia-smi

diff --git a/crates/shivini/src/test.rs b/crates/shivini/src/test.rs
@@ -800,7 +800,7 @@ mod zksync {
 
     pub type ZksyncProof = Proof<F, DefaultTreeHasher, GoldilocksExt2>;
 
-    const TEST_DATA_ROOT_DIR: &str = "./test_data";
+    const TEST_DATA_ROOT_DIR: &str = "./crates/shivini/test_data";
     const DEFAULT_CIRCUIT_INPUT: &str = "default.circuit";
 
     use crate::synthesis_utils::{
@@ -928,7 +928,7 @@ mod zksync {
         let _ctx = ProverContext::create().expect("gpu prover context");
 
         for main_dir in ["base", "leaf", "node", "tip"] {
-            let data_dir = format!("./test_data/{}", main_dir);
+            let data_dir = format!("./crates/shivini/test_data/{}", main_dir);
             dbg!(&data_dir);
             let circuits = scan_directory_for_circuits(&data_dir);
             let reference_proofs = scan_directory_for_proofs(&data_dir);
@@ -1019,7 +1019,7 @@ mod zksync {
         let worker = &Worker::new();
 
         for main_dir in ["base", "leaf", "node", "tip"] {
-            let data_dir = format!("./test_data/{}", main_dir);
+            let data_dir = format!("./crates/shivini/test_data/{}", main_dir);
             dbg!(&data_dir);
             let circuits = scan_directory_for_circuits(&data_dir);
 
@@ -1458,7 +1458,7 @@ mod zksync {
         let _worker = Worker::new();
 
         for main_dir in ["base", "leaf", "node", "tip"] {
-            let data_dir = format!("./test_data/{}", main_dir);
+            let data_dir = format!("./crates/shivini/test_data/{}", main_dir);
             let circuits = scan_directory_for_circuits(&data_dir);
 
             let worker = &Worker::new();
@@ -1557,7 +1557,10 @@ mod zksync {
                 })
                 .collect::<Vec<String>>()
                 .pop()
-                .unwrap_or(format!("./test_data/{}", DEFAULT_CIRCUIT_INPUT))
+                .unwrap_or(format!(
+                    "./crates/shivini/test_data/{}",
+                    DEFAULT_CIRCUIT_INPUT
+                ))
         };
 
         let data = std::fs::read(circuit_file_path).expect("circuit file");