Fix graph test for cuda, add graph tests for ci

elftausend · Feb 7, 2024 · 2ac1c1e · 2ac1c1e
1 parent 3464e5a
commit 2ac1c1e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/.github/workflows/gpu.yml b/.github/workflows/gpu.yml
@@ -70,6 +70,8 @@ jobs:
       run: cargo test --verbose --no-default-features --features opencl,lazy
     - name: Run tests (OpenCL, autograd)
       run: cargo test --verbose --no-default-features --features opencl,autograd
+    - name: Run tests (OpenCL, graph)
+      run: cargo test --verbose --no-default-features --features opencl,graph
 
   test-cuda:
 
@@ -89,6 +91,8 @@ jobs:
       run: cargo test --verbose --no-default-features --features cuda,lazy
     - name: Run tests (CUDA, autograd)
       run: cargo test --verbose --no-default-features --features cuda,autograd
+    - name: Run tests (CUDA, graph)
+      run: cargo test --verbose --no-default-features --features cuda,graph
 
   test-all:
 

diff --git a/Cargo.toml b/Cargo.toml
@@ -49,7 +49,7 @@ min-cl = { git = "https://github.com/elftausend/min-cl", optional = true }
 # min-cl = { version = "0.3.0", optional=true }
 
 [features]
-default = ["cpu", "graph", "cached", "autograd", "lazy", "opencl"]
+default = ["cpu", "graph", "cached", "autograd", "lazy", "opencl", "cuda"]
 # default = ["cpu", "lazy", "static-api", "graph", "autograd", "fork", "serde", "json"]
 
 std = []

diff --git a/tests/graph/graph.rs b/tests/graph/graph.rs
@@ -99,7 +99,7 @@ fn test_graph_cu() -> custos::Result<()> {
             assert_eq!(c.ptr, d.ptr);
             assert_eq!(c.ptr, e.ptr);
         }
-        device.optimize_mem_graph(None).unwrap();
+        device.optimize_mem_graph(&device, None).unwrap();
     }
     Ok(())
 }