diff options
Diffstat (limited to 'guests/keccak/src/lib.rs')
| -rw-r--r-- | guests/keccak/src/lib.rs | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/guests/keccak/src/lib.rs b/guests/keccak/src/lib.rs index 4808442..df15ae2 100644 --- a/guests/keccak/src/lib.rs +++ b/guests/keccak/src/lib.rs @@ -5,9 +5,13 @@ extern crate alloc; #[cfg(feature = "no_std")] use alloc::vec::Vec; -#[cfg(feature = "zkm")] -use zkm_runtime::*; +#[cfg(feature = "sp1")] +use sha3_sp1::{Digest, Keccak256}; +#[cfg(feature = "risc0")] +use sha3_risc0::{Digest, Keccak256}; + +#[cfg(not(any(feature = "zkm", feature = "sp1", feature = "risc0")))] use sha3::{Digest, Keccak256}; #[guests_macro::proving_entrypoint] @@ -15,7 +19,7 @@ pub fn main(secret: Vec<u8>, hash: Vec<u8>) -> bool { #[cfg(feature = "zkm")] let result = zkm_runtime::io::keccak(&secret.as_slice()); - #[cfg(not(any(feature = "zkm")))] + #[cfg(not(feature = "zkm"))] let result = { let mut hasher = Keccak256::new(); hasher.update(secret); |