chore: import upstream snapshot with attribution

2026-07-13 13:36:55 +08:00
commit c8a779b1bb
1887 changed files with 3245738 additions and 0 deletions
@@ -0,0 +1,108 @@
+/*
+ * SPDX-FileCopyrightText: Copyright (c) 1993-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+#ifndef CHECK_MACROS_PLUGIN_H
+#define CHECK_MACROS_PLUGIN_H
+
+#include "NvInfer.h"
+#include "common/cudnnWrapper.h"
+#include "vc/checkMacrosPlugin.h"
+#include <mutex>
+#include <optional>
+#include <sstream>
+#include <string>
+#include <utility>
+
+#ifdef _MSC_VER
+#define FN_NAME __FUNCTION__
+#else
+#define FN_NAME __func__
+#endif
+
+namespace nvinfer1
+{
+namespace plugin
+{
+
+//! \throw CudnnError carrying \p msg, after logging it via \c gLogError.
+//! Parameter semantics match \c nvinfer1::plugin::throwCudaError.
+[[noreturn]] void throwCudnnError(char const* file, char const* function, int32_t line, int32_t status,
+    std::optional<std::string> msg = std::nullopt);
+
+//! \throw CublasError carrying \p msg (or a default message derived from \p status if \p msg is
+//!     \c std::nullopt), after logging it via \c gLogError. An explicit empty string is preserved.
+//! Parameter semantics match \c nvinfer1::plugin::throwCudaError.
+[[noreturn]] void throwCublasError(char const* file, char const* function, int32_t line, int32_t status,
+    std::optional<std::string> msg = std::nullopt);
+
+//! \c TRTException specialization for cuDNN failures.
+class CudnnError : public TRTException
+{
+public:
+    //! \see TRTException::TRTException. Sets the subsystem name to \c "Cudnn".
+    CudnnError(char const* fl, char const* fn, int32_t ln, int32_t stat, std::optional<std::string> msg = std::nullopt)
+        : TRTException(fl, fn, ln, stat, std::move(msg), "Cudnn")
+    {
+    }
+};
+
+//! \c TRTException specialization for cuBLAS failures.
+class CublasError : public TRTException
+{
+public:
+    //! \see TRTException::TRTException. Sets the subsystem name to \c "cuBLAS".
+    CublasError(char const* fl, char const* fn, int32_t ln, int32_t stat, std::optional<std::string> msg = std::nullopt)
+        : TRTException(fl, fn, ln, stat, std::move(msg), "cuBLAS")
+    {
+    }
+};
+
+} // namespace plugin
+
+} // namespace nvinfer1
+
+#define PLUGIN_CHECK_CUDNN(call)                                                                                       \
+    do                                                                                                                 \
+    {                                                                                                                  \
+        cudnnStatus_t status = call;                                                                                   \
+        if (status != CUDNN_STATUS_SUCCESS)                                                                            \
+        {                                                                                                              \
+            return status;                                                                                             \
+        }                                                                                                              \
+    } while (0)
+
+#define PLUGIN_CUBLASASSERT(status_)                                                                                   \
+    {                                                                                                                  \
+        auto s_ = status_;                                                                                             \
+        if (s_ != nvinfer1::pluginInternal::CUBLAS_STATUS_SUCCESS)                                                     \
+        {                                                                                                              \
+            nvinfer1::plugin::throwCublasError(__FILE__, FN_NAME, __LINE__, s_);                                       \
+        }                                                                                                              \
+    }
+
+#define PLUGIN_CUDNNASSERT(status_)                                                                                    \
+    {                                                                                                                  \
+        auto s_ = status_;                                                                                             \
+        if (s_ != CUDNN_STATUS_SUCCESS)                                                                                \
+        {                                                                                                              \
+            nvinfer1::pluginInternal::CudnnWrapper& wrapper                                                            \
+                = nvinfer1::pluginInternal::getCudnnWrapper(/* plugin caller name */ nullptr);                         \
+            const char* msg = wrapper.cudnnGetErrorString(s_);                                                         \
+            nvinfer1::plugin::throwCudnnError(__FILE__, FN_NAME, __LINE__, s_, msg);                                   \
+        }                                                                                                              \
+    }
+
+#endif /*CHECK_MACROS_PLUGIN_H*/