hw: Allow up to 8 proof segment threads

Chia-Network · Oct 13, 2023 · 187830f · 187830f
1 parent cfcb2c1
commit 187830f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/hw/hw_vdf_client.cpp b/src/hw/hw_vdf_client.cpp
@@ -419,6 +419,7 @@ int parse_opts(int argc, char **argv, struct vdf_client_opts *opts)
         {"vdfs-mask", required_argument, NULL, 1},
         {"vdf-threads", required_argument, NULL, 1},
         {"proof-threads", required_argument, NULL, 1},
+        {"segment-threads", required_argument, NULL, 1},
         {"list", no_argument, NULL, 1},
         {"auto-freq-period", required_argument, NULL, 1},
         {0}
@@ -496,6 +497,10 @@ int parse_opts(int argc, char **argv, struct vdf_client_opts *opts)
         LOG_SIMPLE("Number of proof threads must be less than VDF threads");
         return -1;
     }
+    if (opts->vpo.segment_threads > 8) {
+        LOG_SIMPLE("Number of proof threads per segment must be between 1 and 8");
+        return -1;
+    }
     if (opts->auto_freq && opts->auto_freq_period < 10) {
         LOG_SIMPLE("Invalid auto freq period");
         return -1;