fmt

Author: JannikSt

miner/src/cli/command.rs

@@ -285,7 +285,16 @@ pub async fn execute_command(
                 std::process::exit(1);
             };
 
-            match compute_node_ops.add_compute_node().await {
+            let gpu_count: u32 = match &node_config.compute_specs {
+                Some(specs) => specs
+                    .gpu
+                    .as_ref()
+                    .map(|gpu| gpu.count.unwrap_or(0))
+                    .unwrap_or(0),
+                None => 0,
+            };
+
+            match compute_node_ops.add_compute_node(gpu_count).await {
                 Ok(added_node) => {
                     if added_node {
                         // If we are adding a new compute node we wait for a proper

miner/src/operations/compute_node.rs

@@ -29,7 +29,10 @@ impl<'c> ComputeNodeOperations<'c> {
     }
 
     // Returns true if the compute node was added, false if it already exists
-    pub async fn add_compute_node(&self) -> Result<bool, Box<dyn std::error::Error>> {
+    pub async fn add_compute_node(
+        &self,
+        gpu_count: u32,
+    ) -> Result<bool, Box<dyn std::error::Error>> {
         Console::section("🔄 Adding compute node");
         let compute_node = self
             .compute_registry
@@ -77,7 +80,7 @@ impl<'c> ComputeNodeOperations<'c> {
             .as_bytes();
 
         // Create the signature bytes
-        let compute_units: U256 = U256::from(1000);
+        let compute_units: U256 = U256::from(1000 * gpu_count);
         let add_node_tx = self
             .prime_network
             .add_compute_node(node_address, compute_units, signature.to_vec())