fix vec no_std

facebook · Feb 28, 2024 · 2570f56 · 2570f56
1 parent 7c5f0a7
commit 2570f56
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/winterfell/src/lib.rs b/winterfell/src/lib.rs
@@ -579,6 +579,8 @@
 //! * [A Framework for Efficient STARKs](https://medium.com/starkware/a-framework-for-efficient-starks-19608ba06fbe)
 
 #![cfg_attr(not(feature = "std"), no_std)]
+#[cfg(not(feature = "std"))]
+extern crate alloc;
 
 pub use prover::{
     crypto, iterators, math, matrix, Air, AirContext, Assertion, AuxTraceRandElements,

diff --git a/winterfell/src/tests.rs b/winterfell/src/tests.rs
@@ -5,6 +5,9 @@ use prover::{
     matrix::ColMatrix,
 };
 
+#[cfg(not(feature = "std"))]
+use alloc::{vec, vec::Vec};
+
 #[test]
 fn test_lagrange_kernel_air() {
     let trace = LagrangeMockTrace::new();