Merge pull request #1805 from luxonis/dev_mt_isGPUAvailable

Add hasGPU on device test

Author: MaticTonin

bindings/python/src/DeviceBindings.cpp

@@ -1020,6 +1020,13 @@ void DeviceBindings::bind(pybind11::module& m, void* pCallstack) {
                 return d.isNeuralDepthSupported();
             },
             DOC(dai, DeviceBase, isNeuralDepthSupported))
+        .def(
+            "hasGPU",
+            [](DeviceBase& d) {
+                py::gil_scoped_release release;
+                return d.hasGPU();
+            },
+            DOC(dai, DeviceBase, hasGPU))
         .def(
             "getSupportedDeviceModels",
             [](DeviceBase& d) {

cmake/Depthai/DepthaiDeviceRVC4Config.cmake

@@ -3,4 +3,4 @@
 set(DEPTHAI_DEVICE_RVC4_MATURITY "snapshot")
 
 # "version if applicable"
-set(DEPTHAI_DEVICE_RVC4_VERSION "0.0.1+28ccaa69a10aa51f93a00171259772802cf6bd1b")
+set(DEPTHAI_DEVICE_RVC4_VERSION "0.0.1+fbdb9dd70158a009537e2a84a6d1e83f069c79cf")

examples/cpp/CMakeLists.txt

@@ -149,6 +149,7 @@ add_subdirectory(NeuralNetwork)
 add_subdirectory(Remapping)
 add_subdirectory(RecordReplay)
 add_subdirectory(RVC2)
+addsubdirectory(RVC4)
 add_subdirectory(StereoDepth)
 add_subdirectory(Script)
 add_subdirectory(SpatialDetectionNetwork)

examples/cpp/RVC4/CMakeLists.txt

@@ -0,0 +1,5 @@
+project(rvc4_examples)
+cmake_minimum_required(VERSION 3.10)
+
+add_subdirectory(GPUStereo)
+

examples/cpp/RVC4/GPUStereo/CMakeLists.txt

@@ -0,0 +1,5 @@
+project(rvc4_gpu_stereo_examples)
+cmake_minimum_required(VERSION 3.10)
+
+dai_add_example(check_device_gpu check_device_gpu.cpp ON OFF)
+dai_set_example_test_labels(check_device_gpu rvc4)

examples/cpp/RVC4/GPUStereo/check_device_gpu.cpp

@@ -0,0 +1,31 @@
+#include <iostream>
+#include <string>
+
+#include <argparse/argparse.hpp>
+
+#include "depthai/depthai.hpp"
+
+int main(int argc, char** argv) {
+    argparse::ArgumentParser program("check_device_gpu", "1.0.0");
+    program.add_description("Print basic device GPU availability info.");
+    program.add_argument("--device", "-d")
+        .default_value(std::string(""))
+        .help("Device IP address / device ID (default: auto-discover)");
+
+    try {
+        program.parse_args(argc, argv);
+    } catch(const std::runtime_error& err) {
+        std::cerr << err.what() << '\n';
+        std::cerr << program;
+        return EXIT_FAILURE;
+    }
+
+    const auto deviceArg = program.get<std::string>("--device");
+    dai::Device device = deviceArg.empty() ? dai::Device() : dai::Device(deviceArg);
+
+    std::cout << "Product: " << device.getProductName() << std::endl;
+    std::cout << "Platform: " << device.getPlatformAsString() << std::endl;
+    std::cout << "GPU available: " << (device.hasGPU() ? "true" : "false") << std::endl;
+    return 0;
+}
+

examples/python/RVC4/GPUStereo/check_device_gpu.py

@@ -0,0 +1,18 @@
+#!/usr/bin/env python3
+"""Print basic device GPU availability info."""
+
+import argparse
+
+import depthai as dai
+
+
+parser = argparse.ArgumentParser(description="Check device GPU availability")
+parser.add_argument("--device", "-d", type=str, default=None, help="Device IP address (default: auto-discover)")
+args = parser.parse_args()
+
+device = dai.Device(args.device) if args.device else dai.Device()
+
+print("Product:", device.getProductName())
+print("Platform:", device.getPlatformAsString())
+print("GPU available:", device.hasGPU())
+

include/depthai/device/DeviceBase.hpp

@@ -1002,6 +1002,14 @@ class DeviceBase {
      */
     bool isNeuralDepthSupported();
 
+    /**
+     * Checks if a GPU is available on the device.
+     *
+     * @note This is only meaningful on RVC4 platforms.
+     * @returns True if supported, false otherwise
+     */
+    bool hasGPU();
+
     /**
      * Returns the subset of device zoo models currently available on the device.
      *

src/device/DeviceBase.cpp

@@ -1523,6 +1523,11 @@ bool DeviceBase::isNeuralDepthSupported() {
     return pimpl->rpcCallChecked<bool>("isNeuralDepthSupported");
 }
 
+bool DeviceBase::hasGPU() {
+    if(getPlatform() != Platform::RVC4) return false;
+    return pimpl->rpcCallChecked<bool>("hasGPU");
+}
+
 std::vector<DeviceModelZoo> DeviceBase::getSupportedDeviceModels() {
     return pimpl->rpcCallChecked<std::vector<DeviceModelZoo>>("getSupportedDeviceModels");
 }

tests/CMakeLists.txt

@@ -645,6 +645,10 @@ dai_set_test_labels(resolutions_test ondevice) # TODO(jakob) Make the test runna
 dai_add_test(serialization_test src/onhost_tests/serialization_test.cpp)
 dai_set_test_labels(serialization_test ondevice rvc2_all ci)
 
+# GPU availability test
+dai_add_test(gpu_availability_test src/ondevice_tests/gpu_availability_test.cpp)
+dai_set_test_labels(gpu_availability_test ondevice rvc4)
+
 # Subnode test
 dai_add_test(subnode_test src/ondevice_tests/pipeline/subnode_test.cpp)
 dai_set_test_labels(subnode_test ondevice rvc2_all rvc4 rvc4rgb ci)