feat: support #PBS -l ncpus=N directive for resource cost

Author: ForblazeProject

MANUAL.md

@@ -172,7 +172,7 @@ FBQueue automatically maps embedded script directives to internal job parameters
 | Feature | External Notation (Example) | Internal Parameter |
 | :--- | :--- | :--- |
 | **Job Name** | `#PBS -N name`, `#SBATCH -J name` | **`name`** |
-| **Resource/Cost** | `#$ -pe smp 4`, `#SBATCH -c 4` | **`cost`** |
+| **Resource/Cost** | `#$ -pe smp 4`, `#SBATCH -c 4`, `#PBS -l ncpus=4` | **`cost`** |
 | **Target Queue** | `#PBS -q express`, `#SBATCH -p express` | **`queue`** |
 | **Stdout Path** | `#$ -o /path/to/log`, `#SBATCH -o ...` | **`stdout`** |
 | **Stderr Path** | `#PBS -e /path/to/err`, `#SBATCH -e ...` | **`stderr`** |

PBS_COMPATIBILITY.md

@@ -38,6 +38,7 @@ FBQueue respects your existing job scripts. You don't need to rewrite them. It a
 | `#PBS -N <name>` | Sets the job display name |
 | `#PBS -q <queue>` | Routes the job to a specific FBQueue queue |
 | `#PBS -l nodes=1:ppn=N` | Maps `ppn` to the job's resource `cost` |
+| `#PBS -l ncpus=N` | Maps `ncpus` to the job's resource `cost` |
 | `#$ -pe smp N` | Maps `N` to the job's resource `cost` |
 | `#PBS -o <path>` | Redirects standard output |
 | `#PBS -e <path>` | Redirects standard error |

src/job.rs

@@ -133,6 +133,8 @@ pub fn submit_job(cmd_tmpl: &str, args_tmpl: &[String], cwd: &Path, val: Option<
                                     if let Some(pos) = arg.find("ppn=") {
                                         script_cost = arg[pos+4..].parse().unwrap_or(script_cost);
                                     }
+                                } else if arg.starts_with("ncpus=") {
+                                    script_cost = arg[6..].parse().unwrap_or(script_cost);
                                 }
                                 i += 2;
                             }