From 83b2b7077fd0f27dcaaa8cb6afbfd7fb76e9bc14 Mon Sep 17 00:00:00 2001
From: Green Baneling <XgreenX9999@gmail.com>
Date: Sat, 15 Jun 2024 23:53:11 +0200
Subject: [PATCH] Updated VM initialization benchmark to include many inputs
 and outputs (#1973)

Covers new behavior from https://github.com/FuelLabs/fuel-vm/pull/770

### Before requesting review
- [x] I have reviewed the code myself
---
 CHANGELOG.md                         |  3 +++
 benches/benches/vm_initialization.rs | 34 +++++++++++++++++++++++-----
 2 files changed, 31 insertions(+), 6 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 7bbd687934..a21a24f43e 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -6,6 +6,9 @@ and this project adheres to [Semantic Versioning](http://semver.org/).
 
 ## [Unreleased]
 
+### Changed
+- [#1973](https://github.com/FuelLabs/fuel-core/pull/1973): Updated VM initialization benchmark to include many inputs and outputs.
+
 ## [Version 0.29.0]
 
 ### Added
diff --git a/benches/benches/vm_initialization.rs b/benches/benches/vm_initialization.rs
index 93644a762d..a2a5347363 100644
--- a/benches/benches/vm_initialization.rs
+++ b/benches/benches/vm_initialization.rs
@@ -16,6 +16,7 @@ use fuel_core_types::{
         Output,
         Script,
         Transaction,
+        TxParameters,
     },
     fuel_types::canonical::Serialize,
     fuel_vm::{
@@ -39,7 +40,7 @@ fn transaction<R: Rng>(
     script_data: Vec<u8>,
     consensus_params: &ConsensusParameters,
 ) -> Checked<Script> {
-    let inputs = (0..1)
+    let mut inputs = (0..1)
         .map(|_| {
             Input::coin_predicate(
                 rng.gen(),
@@ -52,13 +53,33 @@ fn transaction<R: Rng>(
                 vec![255; 1],
             )
         })
-        .collect();
+        .collect::<Vec<_>>();
 
-    let outputs = (0..1)
+    const CONTRACTS_NUMBER: u16 = 254;
+
+    for i in 0..CONTRACTS_NUMBER {
+        inputs.push(Input::contract(
+            rng.gen(),
+            rng.gen(),
+            rng.gen(),
+            rng.gen(),
+            [i as u8; 32].into(),
+        ));
+    }
+
+    let mut outputs = (0..1)
         .map(|_| {
             Output::variable(Default::default(), Default::default(), Default::default())
         })
-        .collect();
+        .collect::<Vec<_>>();
+
+    for i in 0..CONTRACTS_NUMBER {
+        outputs.push(Output::contract(
+            i + 1,
+            Default::default(),
+            Default::default(),
+        ));
+    }
 
     Transaction::script(
         1_000_000,
@@ -75,12 +96,13 @@ fn transaction<R: Rng>(
 
 pub fn vm_initialization(c: &mut Criterion) {
     let mut rng = StdRng::seed_from_u64(8586);
-    let consensus_params = ConsensusParameters::default();
+    let mut consensus_params = ConsensusParameters::default();
+    consensus_params.set_tx_params(TxParameters::default().with_max_size(256 * 1024));
     let mut group = c.benchmark_group("vm_initialization");
 
     // Increase the size of the script to measure the performance of the VM initialization
     // with a large script. THe largest allowed script is 64 KB = 8 * 2^13 bytes.
-    const TX_SIZE_POWER_OF_TWO: usize = 12;
+    const TX_SIZE_POWER_OF_TWO: usize = 13;
 
     for i in 5..=TX_SIZE_POWER_OF_TWO {
         let size = 8 * (1 << i);