forked from M-Labs/nac3
1
0
Fork 0

standalone: Treat -T0 as using all available threads

This commit is contained in:
David Mak 2023-10-13 14:57:09 +08:00
parent 2c44b58bb8
commit 48eb64403f
1 changed files with 9 additions and 1 deletions

View File

@ -38,7 +38,8 @@ struct CommandLineArgs {
/// The name of the input file. /// The name of the input file.
file_name: String, file_name: String,
/// The number of threads allocated to processing the source file. /// The number of threads allocated to processing the source file. If 0 is passed to this
/// parameter, all available threads will be used for compilation.
#[arg(short = 'T', default_value_t = 1)] #[arg(short = 'T', default_value_t = 1)]
threads: u32, threads: u32,
@ -207,6 +208,13 @@ fn main() {
.map(|arg| if arg == "native" { host_target_machine.cpu.clone() } else { arg }) .map(|arg| if arg == "native" { host_target_machine.cpu.clone() } else { arg })
.unwrap_or_default(); .unwrap_or_default();
let target_features = target_features.unwrap_or_default(); let target_features = target_features.unwrap_or_default();
let threads = if threads == 0 {
std::thread::available_parallelism()
.map(|threads| threads.get() as u32)
.unwrap_or(1u32)
} else {
threads
};
let opt_level = match opt_level { let opt_level = match opt_level {
0 => OptimizationLevel::None, 0 => OptimizationLevel::None,
1 => OptimizationLevel::Less, 1 => OptimizationLevel::Less,