Merge pull request #66 from sebwolf-de/main

Add priority to hq command
UM-Bridge · Mar 12, 2024 · 2d12741 · 2d12741
2 parents 1e357b0 + 510a4ed
commit 2d12741
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/hpc/LoadBalancer.cpp b/hpc/LoadBalancer.cpp
@@ -41,6 +41,8 @@ const std::vector<std::string> get_model_names() {
     return umbridge::SupportedModels(hq_job.server_url);
 }
 
+std::atomic<int32_t> HyperQueueJob::job_count = 0;
+
 int main(int argc, char *argv[])
 {
     create_directory_if_not_existing("urls");

diff --git a/hpc/LoadBalancer.hpp b/hpc/LoadBalancer.hpp
@@ -71,6 +71,7 @@ int hq_submit_delay_ms = 0;
 class HyperQueueJob
 {
 public:
+    static std::atomic<int32_t> job_count;
     HyperQueueJob(std::string model_name, bool start_client=true, 
                                           bool force_default_submission_script=false)
     {
@@ -113,6 +114,7 @@ class HyperQueueJob
         const std::filesystem::path submission_script_model_specific("job_" + model_name + ".sh");
 
         std::string hq_command = "hq submit --output-mode=quiet ";
+        hq_command += "--priority=" + std::to_string(job_count) + " ";
         if (std::filesystem::exists(submission_script_dir / submission_script_model_specific) && !force_default_submission_script)
         {
             hq_command += (submission_script_dir / submission_script_model_specific).string();
@@ -128,6 +130,7 @@ class HyperQueueJob
 
         // Submit the HQ job and retrieve the HQ job ID.
         std::string job_id = getCommandOutput(hq_command);
+        job_count--;
 
         // Delete the line break.
         if (!job_id.empty())