Thomas 2 lat temu
rodzic
commit
328f0316ba
1 zmienionych plików z 5 dodań i 1 usunięć
  1. 5 1
      src/main.rs

+ 5 - 1
src/main.rs

@@ -27,6 +27,10 @@ struct Args {
     /// output tsv path
     #[arg(short, long)]
     genome: String,
+
+    /// output tsv path
+    #[arg(short, long)]
+    threads: usize,
 }
 
 fn main() {
@@ -64,7 +68,7 @@ fn main() {
 
     let n_chunks = ranges.len() / chunk_size;
 
-    ThreadPoolBuilder::new().num_threads(15).build_global().unwrap();
+    ThreadPoolBuilder::new().num_threads(args.threads).build_global().unwrap();
 
     ranges.par_chunks(chunk_size)
     .progress_count(n_chunks as u64)