added gpu support for docker platform

libpod/src/db_model.rs

@@ -51,7 +51,7 @@ pub struct PluginRunItem {
     pub execute_on: PluginExecutionPlatform,
 }
 
-#[derive(Deserialize, Debug, Default)]
+#[derive(Deserialize, Debug, Default, PartialEq)]
 #[serde(rename_all = "camelCase")]
 pub enum PluginExecutionPlatform {
     #[default]

libpod/src/plugin_run.rs

@@ -211,6 +211,10 @@ fn run_docker_container(
         args.extend_from_slice(&["--label".to_string(), format!("{UNIQUE_ID}={unique_id}")]);
     }
 
+    if plugin.execute_on == PluginExecutionPlatform::Gpu {
+        args.extend_from_slice(&["--gpus=all".to_string()]);
+    }
+
     args.extend_from_slice(&["--".to_string(), plugin.container_image.clone()]);
 
     let envs: HashMap<&str, &str> = HashMap::new();