Merge pull request #20 from eth-applied-research-group/kw/r0-gpu

feat: Add GPU support for risc0

crates/ere-risczero/Cargo.toml

@@ -17,5 +17,9 @@ tempfile = "3.3"
 serde_json = "1.0"
 thiserror = "2"
 
+[features]
+metal = ["risc0-zkvm/metal"]
+cuda = ["risc0-zkvm/cuda"]
+
 [lints]
 workspace = true

crates/ere-risczero/src/lib.rs

@@ -29,6 +29,17 @@ impl EreRisc0 {
         program: <RV32_IM_RISCZERO_ZKVM_ELF as Compiler>::Program,
         resource_type: ProverResourceType,
     ) -> Self {
+        match resource_type {
+            ProverResourceType::Cpu => {
+                #[cfg(any(feature = "cuda", feature = "metal"))]
+                panic!("CPU mode requires both 'cuda' and 'metal' features to be disabled");
+            }
+            ProverResourceType::Gpu => {
+                #[cfg(not(any(feature = "cuda", feature = "metal")))]
+                panic!("GPU selected but neither 'cuda' nor 'metal' feature is enabled");
+            }
+        }
+
         Self {
             program,
             resource_type,