From 7ff15e15e08d0d8603633980bb128f447861c947 Mon Sep 17 00:00:00 2001
From: Carl Hetherington <cth@carlh.net>
Date: Mon, 3 Mar 2025 23:29:06 +0100
Subject: Add list-gpus command to dcpomatic2_cli.

---
 src/lib/encode_cli.cc | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

(limited to 'src/lib/encode_cli.cc')

diff --git a/src/lib/encode_cli.cc b/src/lib/encode_cli.cc
index f28b93421..7681252b2 100644
--- a/src/lib/encode_cli.cc
+++ b/src/lib/encode_cli.cc
@@ -30,6 +30,7 @@
 #include "filter.h"
 #ifdef DCPOMATIC_GROK
 #include "grok/context.h"
+#include "grok/util.h"
 #endif
 #include "hints.h"
 #include "job_manager.h"
@@ -75,6 +76,7 @@ help(function <void (string)> out)
 #ifdef DCPOMATIC_GROK
 	out("  config-params                list the parameters that can be set with `config`\n");
 	out("  config <PARAMETER> <VALUE>   set a DCP-o-matic configuration value\n");
+	out("  list-gpus                    list available GPUs\n");
 #endif
 
 	out("\nOptions:\n\n");
@@ -382,7 +384,8 @@ encode_cli(int argc, char* argv[], function<void (string)> out, function<void ()
 #ifdef DCPOMATIC_GROK
 		"dump",
 		"config-params",
-		"config"
+		"config",
+		"list-gpus"
 #else
 		"dump"
 #endif
@@ -434,6 +437,12 @@ encode_cli(int argc, char* argv[], function<void (string)> out, function<void ()
 			return fmt::format("Missing configuration parameter: use {} config <parameter> <value>", program_name);
 		}
 		return {};
+	} else if (command == "list-gpus") {
+		int N = 0;
+		for (auto gpu: get_gpu_names()) {
+			out(fmt::format("{}: {}\n", N++, gpu));
+		}
+		return {};
 	}
 #endif
 
-- 
cgit v1.2.3