switch wee_alloc for rlsf

anoma · May 30, 2024 · bfbecad · bfbecad
1 parent 5a5838a
commit bfbecad
Show file tree

Hide file tree

Showing 67 changed files with 263 additions and 225 deletions.
diff --git a/crates/macros/src/lib.rs b/crates/macros/src/lib.rs
@@ -28,9 +28,9 @@ pub fn transaction(_attr: TokenStream, input: TokenStream) -> TokenStream {
     let ast = parse_macro_input!(input as ItemFn);
     let ident = &ast.sig.ident;
     let gen = quote! {
-        // Use `wee_alloc` as the global allocator.
+        // Use `rlsf` as the global allocator.
         #[global_allocator]
-        static ALLOC: wee_alloc::WeeAlloc = wee_alloc::WeeAlloc::INIT;
+        static ALLOC: rlsf::SmallGlobalTlsf = rlsf::SmallGlobalTlsf::new();
 
         #ast
 
@@ -88,9 +88,9 @@ pub fn validity_predicate(
     let ast = parse_macro_input!(input as ItemFn);
     let ident = &ast.sig.ident;
     let gen = quote! {
-        // Use `wee_alloc` as the global allocator.
+        // Use `rlsf` as the global allocator.
         #[global_allocator]
-        static ALLOC: wee_alloc::WeeAlloc = wee_alloc::WeeAlloc::INIT;
+        static ALLOC: rlsf::SmallGlobalTlsf = rlsf::SmallGlobalTlsf::new();
 
         #ast