tx wide compute budget

Cargo.toml

@@ -43,6 +43,7 @@ members = [
     "poh-bench",
     "program-test",
     "programs/bpf_loader",
+    "programs/compute-budget",
     "programs/config",
     "programs/failure",
     "programs/noop",

programs/bpf/tests/programs.rs

@@ -34,6 +34,7 @@ use solana_sdk::{
     bpf_loader, bpf_loader_deprecated, bpf_loader_upgradeable,
     client::SyncClient,
     clock::MAX_PROCESSING_AGE,
+    compute_budget,
     entrypoint::{MAX_PERMITTED_DATA_INCREASE, SUCCESS},
     instruction::{AccountMeta, CompiledInstruction, Instruction, InstructionError},
     keyed_account::KeyedAccount,
@@ -1234,8 +1235,6 @@ fn test_program_bpf_call_depth() {
 
     solana_logger::setup();
 
-    println!("Test program: solana_bpf_rust_call_depth");
-
     let GenesisConfigInfo {
         genesis_config,
         mint_keypair,
@@ -1269,6 +1268,40 @@ fn test_program_bpf_call_depth() {
     assert!(result.is_err());
 }
 
+#[cfg(feature = "bpf_rust")]
+#[test]
+fn test_program_bpf_compute_budget() {
+    solana_logger::setup();
+
+    let GenesisConfigInfo {
+        genesis_config,
+        mint_keypair,
+        ..
+    } = create_genesis_config(50);
+    let mut bank = Bank::new(&genesis_config);
+    let (name, id, entrypoint) = solana_bpf_loader_program!();
+    bank.add_builtin(&name, id, entrypoint);
+    let bank_client = BankClient::new(bank);
+    let program_id = load_bpf_program(
+        &bank_client,
+        &bpf_loader::id(),
+        &mint_keypair,
+        "solana_bpf_rust_noop",
+    );
+    let message = Message::new(
+        &[
+            compute_budget::request_units(1),
+            Instruction::new_with_bincode(program_id, &0, vec![]),
+        ],
+        Some(&mint_keypair.pubkey()),
+    );
+    let result = bank_client.send_and_confirm_message(&[&mint_keypair], message);
+    assert_eq!(
+        result.unwrap_err().unwrap(),
+        TransactionError::InstructionError(1, InstructionError::ProgramFailedToComplete),
+    );
+}
+
 #[test]
 fn assert_instruction_count() {
     solana_logger::setup();

programs/compute-budget/Cargo.toml

@@ -0,0 +1,20 @@
+[package]
+name = "solana-compute-budget-program"
+description = "Solana Compute Budget program"
+version = "1.8.0"
+homepage = "https://solana.com/"
+documentation = "https://docs.rs/solana-compute-budget-program"
+repository = "https://github.com/solana-labs/solana"
+authors = ["Solana Maintainers <[email protected]>"]
+license = "Apache-2.0"
+edition = "2018"
+
+[dependencies]
+solana-sdk = { path = "../../sdk", version = "=1.8.0" }
+
+[lib]
+crate-type = ["lib"]
+name = "solana_compute_budget_program"
+
+[package.metadata.docs.rs]
+targets = ["x86_64-unknown-linux-gnu"]

programs/compute-budget/src/lib.rs

@@ -0,0 +1,12 @@
+use solana_sdk::{
+    instruction::InstructionError, process_instruction::InvokeContext, pubkey::Pubkey,
+};
+
+pub fn process_instruction(
+    _program_id: &Pubkey,
+    _data: &[u8],
+    _invoke_context: &mut dyn InvokeContext,
+) -> Result<(), InstructionError> {
+    // Do nothing, compute budget instructions handled by the runtime
+    Ok(())
+}

runtime/Cargo.toml

@@ -37,6 +37,7 @@ regex = "1.5.4"
 serde = { version = "1.0.126", features = ["rc"] }
 serde_derive = "1.0.103"
 solana-config-program = { path = "../programs/config", version = "=1.8.0" }
+solana-compute-budget-program = { path = "../programs/compute-budget", version = "=1.8.0" }
 solana-frozen-abi = { path = "../frozen-abi", version = "=1.8.0" }
 solana-frozen-abi-macro = { path = "../frozen-abi/macro", version = "=1.8.0" }
 solana-logger = { path = "../logger", version = "=1.8.0" }