blockifier: convert builtins_to_gas to cairo_primitives_to_gas

Author: avivg-starkware

crates/blockifier/src/bouncer.rs

@@ -16,6 +16,7 @@ use crate::blockifier::transaction_executor::{
 };
 use crate::blockifier_versioned_constants::{BuiltinGasCosts, VersionedConstants};
 use crate::execution::call_info::{
+    cairo_primitive_counter_map,
     BuiltinCounterMap,
     CairoPrimitiveCounterMap,
     CairoPrimitiveName,
@@ -688,7 +689,12 @@ fn proving_gas_from_builtins_and_sierra_gas(
     proving_builtin_gas_costs: &BuiltinGasCosts,
     sierra_builtin_gas_costs: &BuiltinGasCosts,
 ) -> GasAmount {
-    let builtins_proving_gas = builtins_to_gas(builtin_counters, proving_builtin_gas_costs);
+    // TODO(AvivG): To support opcodes in the computation, pass cairo_primitive_counter_map to this
+    // function
+    let builtins_proving_gas = cairo_primitives_to_gas(
+        &cairo_primitive_counter_map(builtin_counters.iter().map(|(k, v)| (*k, *v))),
+        proving_builtin_gas_costs,
+    );
     let steps_proving_gas =
         sierra_gas_to_steps_gas(sierra_gas, builtin_counters, sierra_builtin_gas_costs);
 
@@ -701,7 +707,12 @@ pub fn vm_resources_to_gas(
     builtin_gas_cost: &BuiltinGasCosts,
     versioned_constants: &VersionedConstants,
 ) -> GasAmount {
-    let builtins_gas_cost = builtins_to_gas(&resources.prover_builtins(), builtin_gas_cost);
+    // TODO(AvivG): To support opcodes in the computation, pass cairo_primitive_counter_map to this
+    // function
+    let builtins_gas_cost = cairo_primitives_to_gas(
+        &cairo_primitive_counter_map(resources.prover_builtins()),
+        builtin_gas_cost,
+    );
     let n_steps_gas_cost = n_steps_to_gas(resources.total_n_steps(), versioned_constants);
     let n_memory_holes_gas_cost =
         memory_holes_to_gas(resources.n_memory_holes, versioned_constants);
@@ -717,7 +728,12 @@ pub fn sierra_gas_to_steps_gas(
     builtin_counters: &BuiltinCounterMap,
     sierra_builtin_gas_costs: &BuiltinGasCosts,
 ) -> GasAmount {
-    let builtins_gas_cost = builtins_to_gas(builtin_counters, sierra_builtin_gas_costs);
+    // TODO(AvivG): To support opcodes in the computation, pass cairo_primitive_counter_map to this
+    // function
+    let builtins_gas_cost = cairo_primitives_to_gas(
+        &cairo_primitive_counter_map(builtin_counters.clone()),
+        sierra_builtin_gas_costs,
+    );
 
     sierra_gas.checked_sub(builtins_gas_cost).unwrap_or_else(|| {
         log::debug!(
@@ -728,24 +744,29 @@ pub fn sierra_gas_to_steps_gas(
     })
 }
 
-pub fn builtins_to_gas(
-    builtin_counters: &BuiltinCounterMap,
-    builtin_gas_costs: &BuiltinGasCosts,
+pub fn cairo_primitives_to_gas(
+    cairo_primitives_counters: &CairoPrimitiveCounterMap,
+    // NOTE: 'blake' is currently the only supported opcode, by being included in the
+    // builtin_gas_costs.
+    cairo_primitives_gas_costs: &BuiltinGasCosts,
 ) -> GasAmount {
-    let builtin_gas = builtin_counters.iter().fold(0u64, |accumulated_gas, (name, &count)| {
-        let builtin_weight = builtin_gas_costs.get_builtin_gas_cost(name).unwrap();
-        builtin_weight
-            .checked_mul(u64_from_usize(count))
-            .and_then(|builtin_gas| accumulated_gas.checked_add(builtin_gas))
-            .unwrap_or_else(|| {
-                panic!(
-                    "Overflow while converting builtin counters to gas.\nBuiltin: {name}, Weight: \
-                     {builtin_weight}, Count: {count}, Accumulated gas: {accumulated_gas}"
-                )
-            })
-    });
+    let cairo_primitives_gas =
+        cairo_primitives_counters.iter().fold(0u64, |accumulated_gas, (name, &count)| {
+            let cairo_primitive_weight =
+                cairo_primitives_gas_costs.get_cairo_primitive_gas_cost(&name).unwrap();
+            cairo_primitive_weight
+                .checked_mul(u64_from_usize(count))
+                .and_then(|builtin_gas| accumulated_gas.checked_add(builtin_gas))
+                .unwrap_or_else(|| {
+                    panic!(
+                        "Overflow while converting cairo primitives counters to gas.\nCairo \
+                         primitive: {name}, Weight: {cairo_primitive_weight}, Count: {count}, \
+                         Accumulated gas: {accumulated_gas}"
+                    )
+                })
+        });
 
-    GasAmount(builtin_gas)
+    GasAmount(cairo_primitives_gas)
 }
 
 fn add_casm_hash_computation_gas_cost(

crates/blockifier/src/bouncer_test.rs

@@ -16,7 +16,7 @@ use starknet_api::{class_hash, contract_address, felt, storage_key};
 use super::BouncerConfig;
 use crate::blockifier::transaction_executor::TransactionExecutorError;
 use crate::bouncer::{
-    builtins_to_gas,
+    cairo_primitives_to_gas,
     get_patricia_update_resources,
     get_tx_weights,
     map_class_hash_to_casm_hash_computation_resources,
@@ -299,8 +299,10 @@ fn test_bouncer_try_update_gas_based(#[case] scenario: &'static str, block_conte
         _ => panic!("Unexpected scenario: {scenario}"),
     };
 
-    let proving_gas_max_capacity =
-        builtins_to_gas(&max_capacity_builtin_counters, &builtin_weights.gas_costs);
+    let proving_gas_max_capacity = cairo_primitives_to_gas(
+        &cairo_primitive_counter_map(max_capacity_builtin_counters),
+        &builtin_weights.gas_costs,
+    );
 
     let block_max_capacity = BouncerWeights {
         l1_gas: 20,