fix build

Author: irexyc

csrc/mmdeploy/apis/CMakeLists.txt

@@ -9,4 +9,3 @@ add_subdirectory(java)
 if (MMDEPLOY_BUILD_SDK_PYTHON_API)
     add_subdirectory(python)
 endif ()
-

csrc/mmdeploy/device/cuda/cuda_device.h

@@ -1,5 +1,8 @@
 // Copyright (c) OpenMMLab. All rights reserved.
 
+#ifndef MMDEPLOY_SRC_DEVICE_CUDA_CUDE_DEVICE_H_
+#define MMDEPLOY_SRC_DEVICE_CUDA_CUDE_DEVICE_H_
+
 #include <any>
 #include <mutex>
 
@@ -196,3 +199,5 @@ class CudaDeviceGuard {
 };
 
 }  // namespace mmdeploy::framework
+
+#endif  // MMDEPLOY_SRC_DEVICE_CUDA_CUDE_DEVICE_H_

csrc/mmdeploy/net/trt/trt_net.cpp

@@ -8,7 +8,6 @@
 #include "mmdeploy/core/model.h"
 #include "mmdeploy/core/module.h"
 #include "mmdeploy/core/utils/formatter.h"
-#include "mmdeploy/device/cuda/cuda_device.h"
 
 namespace mmdeploy::framework {
 

demo/python/to_triton_model.py

@@ -38,9 +38,7 @@ def parse_args():
         dict(name='mask', dtype=pb.TYPE_INT32, dims=[-1, -1]),
         dict(name='score', dtype=pb.TYPE_FP32, dims=[-1, -1, -1])
     ],
-    Restorer=[
-        dict(name='output', dtype=pb.TYPE_FP32, dims=[-1, -1, 3])
-    ],
+    Restorer=[dict(name='output', dtype=pb.TYPE_FP32, dims=[-1, -1, 3])],
     TextDetector=[],
     TextRecognizer=[],
     PoseDetector=[],