[benchmarks] Module loading benches (simple)

aptos-labs · Nov 13, 2024 · 3d9f8f1 · 3d9f8f1
1 parent 2f3708e
commit 3d9f8f1
Show file tree

Hide file tree

Showing 9 changed files with 7,783 additions and 14 deletions.
diff --git a/crates/transaction-generator-lib/src/args.rs b/crates/transaction-generator-lib/src/args.rs
@@ -73,6 +73,9 @@ pub enum TransactionTypeArg {
     SmartTablePicture1MWith1KChangeExceedsLimit,
     DeserializeU256,
     SimpleScript,
+    ChainDependencies,
+    ChainFriends,
+    StarDependencies,
 }
 
 impl TransactionTypeArg {
@@ -320,6 +323,9 @@ impl TransactionTypeArg {
             },
             TransactionTypeArg::DeserializeU256 => call_custom_module(EntryPoints::DeserializeU256),
             TransactionTypeArg::SimpleScript => call_custom_module(EntryPoints::SimpleScript),
+            TransactionTypeArg::ChainDependencies => call_custom_module(EntryPoints::ChainDependencies),
+            TransactionTypeArg::ChainFriends => call_custom_module(EntryPoints::ChainFriends),
+            TransactionTypeArg::StarDependencies => call_custom_module(EntryPoints::StarDependencies),
         }
     }
 

diff --git a/crates/transaction-generator-lib/src/publishing/module_simple.rs b/crates/transaction-generator-lib/src/publishing/module_simple.rs
@@ -272,6 +272,9 @@ pub enum EntryPoints {
     /// there to slow down deserialization & verification, effectively making it more expensive to
     /// load it into code cache.
     SimpleScript,
+    ChainDependencies,
+    ChainFriends,
+    StarDependencies,
 }
 
 impl EntryPoints {
@@ -329,6 +332,9 @@ impl EntryPoints {
             EntryPoints::IncGlobalMilestoneAggV2 { .. }
             | EntryPoints::CreateGlobalMilestoneAggV2 { .. } => "aggregator_examples",
             EntryPoints::DeserializeU256 => "bcs_stream",
+            EntryPoints::ChainDependencies => "module_loading_chain_dependencies",
+            EntryPoints::ChainFriends => "module_loading_chain_friends",
+            EntryPoints::StarDependencies => "module_loading_star_dependencies",
         }
     }
 
@@ -389,6 +395,9 @@ impl EntryPoints {
             EntryPoints::IncGlobalMilestoneAggV2 { .. }
             | EntryPoints::CreateGlobalMilestoneAggV2 { .. } => "counter_with_milestone",
             EntryPoints::DeserializeU256 => "bcs_stream",
+            EntryPoints::ChainDependencies => "module_loading_chain_dependencies",
+            EntryPoints::ChainFriends => "module_loading_chain_friends",
+            EntryPoints::StarDependencies => "module_loading_star_dependencies",
         }
     }
 
@@ -732,6 +741,7 @@ impl EntryPoints {
                     ],
                 )
             },
+            EntryPoints::ChainDependencies | EntryPoints::ChainFriends | EntryPoints::StarDependencies => get_payload_void(module_id, ident_str!("run").to_owned()),
         }
     }
 
@@ -840,6 +850,7 @@ impl EntryPoints {
             EntryPoints::DeserializeU256 => AutomaticArgs::None,
             EntryPoints::IncGlobalMilestoneAggV2 { .. } => AutomaticArgs::None,
             EntryPoints::CreateGlobalMilestoneAggV2 { .. } => AutomaticArgs::Signer,
+            EntryPoints::ChainDependencies | EntryPoints::ChainFriends | EntryPoints::StarDependencies => AutomaticArgs::None,
         }
     }
 }

diff --git a/crates/transaction-generator-lib/src/publishing/raw_module_data.rs b/crates/transaction-generator-lib/src/publishing/raw_module_data.rs
diff --git a/testsuite/module-publish/src/packages/module_loading_chain_dependencies/Move.toml b/testsuite/module-publish/src/packages/module_loading_chain_dependencies/Move.toml
@@ -0,0 +1,9 @@
+[package]
+name = 'module_loading_chain_dependencies'
+version = '1.0.0'
+
+[dependencies]
+AptosStdlib = { local = "../../../../../aptos-move/framework/aptos-stdlib" }
+
+[addresses]
+publisher_address = "0xABCD"