[scheduler] Fix memory values on env vars (AcademySoftwareFoundation#2163)

Environment variables should display memory values in KBs

Author: DiegoTavares

rust/crates/scheduler/src/pipeline/dispatcher/actor.rs

@@ -881,11 +881,14 @@ impl RqdDispatcherService {
         let mut environment = proc.frame.env.clone();
         environment.insert("CUE3".to_string(), "1".to_string());
         environment.insert("CUE_THREADS".to_string(), threads.to_string());
-        environment.insert("CUE_MEMORY".to_string(), proc.memory_reserved.to_string());
+        environment.insert(
+            "CUE_MEMORY".to_string(),
+            (proc.memory_reserved.as_u64() / KIB).to_string(),
+        );
         environment.insert("CUE_GPUS".to_string(), proc.gpus_reserved.to_string());
         environment.insert(
             "CUE_GPU_MEMORY".to_string(),
-            proc.gpu_memory_reserved.to_string(),
+            (proc.gpu_memory_reserved.as_u64() / KIB).to_string(),
         );
         environment.insert("CUE_LOG_PATH".to_string(), frame.log_dir.clone());
         environment.insert("CUE_RANGE".to_string(), frame.range.clone());
@@ -1020,7 +1023,7 @@ mod tests {
 
     use super::*;
     use crate::models::{CoreSize, DispatchFrame, Host};
-    use bytesize::ByteSize;
+    use bytesize::{ByteSize, KB};
     use opencue_proto::host::ThreadMode;
     use uuid::Uuid;
 
@@ -1480,7 +1483,7 @@ mod tests {
         assert_eq!(run_frame.environment.get("CUE_THREADS").unwrap(), "2");
         assert_eq!(
             run_frame.environment.get("CUE_MEMORY").unwrap(),
-            &virtual_proc.memory_reserved.to_string()
+            &(virtual_proc.memory_reserved.as_u64() / KIB).to_string(),
         );
         assert_eq!(run_frame.environment.get("CUE_GPUS").unwrap(), "1");
         assert_eq!(