Added function to set GPU min and max clocks (#16)

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "nvidia_oc"
-version = "0.1.15"
+version = "0.1.16"
 edition = "2021"
 license = "MIT"
 description = "A simple command line tool to overclock Nvidia GPUs using the NVML library on Linux. This supports both X11 and Wayland."

README.md

@@ -7,7 +7,7 @@ NVIDIA_OC is a simple Rust CLI tool designed to overclock NVIDIA GPUs on Linux.
 To set the overclock parameters for your NVIDIA GPU, use the following command:
 
 ```bash
-./nvidia_oc set --index 0 --power-limit 200000 --freq-offset 160 --mem-offset 850
+./nvidia_oc set --index 0 --power-limit 200000 --freq-offset 160 --mem-offset 850 --min-clock 0 --max-clock 2000
 ```
 
 ## Run on Startup
@@ -24,7 +24,7 @@ Description=NVIDIA Overclocking Service
 After=network.target
 
 [Service]
-ExecStart=[path_to_binary]/nvidia_oc set --index 0 --power-limit 200000 --freq-offset 160 --mem-offset 850
+ExecStart=[path_to_binary]/nvidia_oc set --index 0 --power-limit 200000 --freq-offset 160 --mem-offset 850 --min-clock 0 --max-clock 2000
 User=root
 Restart=on-failure
 

example_config.json

@@ -3,7 +3,9 @@
     "0": {
       "freqOffset": 200000,
       "memOffset": 160,
-      "powerLimit": 500
+      "powerLimit": 500,
+      "minClock": 0,
+      "maxClock": 2000
     }
   }
 }

src/main.rs

@@ -42,11 +42,17 @@ struct Sets {
     #[arg(short, long)]
     freq_offset: Option<i32>,
     /// GPU memory frequency offset
-    #[arg(short, long)]
+    #[arg(long = "mem-offset")]
     mem_offset: Option<i32>,
     /// GPU power limit in milliwatts
     #[arg(short, long)]
     power_limit: Option<u32>,
+    /// GPU min clock
+    #[arg(long = "min-clock")]
+    min_clock: Option<u32>,
+    /// GPU max clock
+    #[arg(long = "max-clock")]
+    max_clock: Option<u32>,
 }
 
 impl Sets {
@@ -64,6 +70,11 @@ impl Sets {
         if let Some(limit) = self.power_limit {
             set_gpu_power_limit(&nvml, device, limit).expect("Failed to set GPU power limit");
         }
+
+        if let (Some(min_clock), Some(max_clock)) = (self.min_clock, self.max_clock) {
+            set_gpu_min_max_clock(&nvml, device, min_clock, max_clock)
+                .expect("Failed to set GPU min and max clocks");
+        }
     }
 }
 
@@ -179,6 +190,20 @@ fn set_gpu_power_limit(nvml_lib: &NvmlLib, handle: nvmlDevice_t, limit: u32) ->
     }
 }
 
+fn set_gpu_min_max_clock(
+    nvml_lib: &NvmlLib,
+    handle: nvmlDevice_t,
+    minclock: u32,
+    maxclock: u32,
+) -> Result<(), String> {
+    let result = unsafe { nvml_lib.nvmlDeviceSetGpuLockedClocks(handle, minclock, maxclock) };
+    if result != 0 {
+        Err(format!("Error code: {}", result))
+    } else {
+        Ok(())
+    }
+}
+
 fn generate_completion_script<G: Generator>(gen: G) {
     let mut cmd = Cli::command();
     let name = cmd.get_name().to_string();