Michal Guzek
c8a50438f6
* Updated submodule onnx to 10.5 Signed-off-by: Michal Guzek <mguzek@nvidia.com> * Updated files to 10.5 Signed-off-by: Michal Guzek <mguzek@nvidia.com> * Restore the previous .gitmodules Signed-off-by: Michal Guzek <mguzek@nvidia.com> * .gitmodules new line Signed-off-by: Michal Guzek <mguzek@nvidia.com> * Update public docker files Signed-off-by: Michal Guzek <mguzek@nvidia.com> * Update 10.5 oss deleted files Signed-off-by: Michal Guzek <mguzek@nvidia.com> * Update deprecated APIs Signed-off-by: Michal Guzek <mguzek@nvidia.com> --------- Signed-off-by: Michal Guzek <mguzek@nvidia.com>
123 lines
3.7 KiB
C++
123 lines
3.7 KiB
C++
/*
|
|
* SPDX-FileCopyrightText: Copyright (c) 1993-2024 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.
|
|
*/
|
|
|
|
#include "vc/checkMacrosPlugin.h"
|
|
#include "vc/vfcCommon.h"
|
|
#include <cstdlib>
|
|
#include <cuda_runtime.h>
|
|
|
|
namespace nvinfer1::plugin
|
|
{
|
|
|
|
// This will be populated by the logger supplied by the user to initLibNvInferPlugins()
|
|
ILogger* gLogger{};
|
|
|
|
template <ILogger::Severity tSeverity>
|
|
int32_t LogStream<tSeverity>::Buf::sync()
|
|
{
|
|
std::string s = str();
|
|
while (!s.empty() && s.back() == '\n')
|
|
{
|
|
s.pop_back();
|
|
}
|
|
if (gLogger != nullptr)
|
|
{
|
|
gLogger->log(tSeverity, s.c_str());
|
|
}
|
|
str("");
|
|
return 0;
|
|
}
|
|
|
|
// These use gLogger, and therefore require initLibNvInferPlugins() to be called with a logger
|
|
// (otherwise, it will not log)
|
|
LogStream<ILogger::Severity::kERROR> gLogError;
|
|
LogStream<ILogger::Severity::kWARNING> gLogWarning;
|
|
LogStream<ILogger::Severity::kINFO> gLogInfo;
|
|
LogStream<ILogger::Severity::kVERBOSE> gLogVerbose;
|
|
|
|
// break-pointable
|
|
void throwCudaError(char const* file, char const* function, int32_t line, int32_t status, char const* msg)
|
|
{
|
|
CudaError error(file, function, line, status, msg);
|
|
error.log(gLogError);
|
|
// NOLINTNEXTLINE(misc-throw-by-value-catch-by-reference)
|
|
throw error;
|
|
}
|
|
|
|
// break-pointable
|
|
void throwPluginError(char const* file, char const* function, int32_t line, int32_t status, char const* msg)
|
|
{
|
|
PluginError error(file, function, line, status, msg);
|
|
reportValidationFailure(msg, file, line);
|
|
// NOLINTNEXTLINE(misc-throw-by-value-catch-by-reference)
|
|
throw error;
|
|
}
|
|
|
|
void logError(char const* msg, char const* file, char const* fn, int32_t line)
|
|
{
|
|
gLogError << "Parameter check failed at: " << file << "::" << fn << "::" << line;
|
|
gLogError << ", condition: " << msg << std::endl;
|
|
}
|
|
|
|
void reportValidationFailure(char const* msg, char const* file, int32_t line)
|
|
{
|
|
std::ostringstream stream;
|
|
stream << "Validation failed: " << msg << "\n" << file << ':' << line << "\n";
|
|
#ifdef COMPILE_VFC_PLUGIN
|
|
ILogger* logger = getPluginLogger();
|
|
if (logger != nullptr)
|
|
{
|
|
logger->log(nvinfer1::ILogger::Severity::kINTERNAL_ERROR, stream.str().c_str());
|
|
}
|
|
#else
|
|
getLogger()->log(nvinfer1::ILogger::Severity::kINTERNAL_ERROR, stream.str().c_str());
|
|
#endif
|
|
}
|
|
|
|
// break-pointable
|
|
void reportAssertion(char const* msg, char const* file, int32_t line)
|
|
{
|
|
std::ostringstream stream;
|
|
stream << "Assertion failed: " << msg << "\n"
|
|
<< file << ':' << line << "\n"
|
|
<< "Aborting..."
|
|
<< "\n";
|
|
#ifdef COMPILE_VFC_PLUGIN
|
|
ILogger* logger = getPluginLogger();
|
|
if (logger != nullptr)
|
|
{
|
|
logger->log(nvinfer1::ILogger::Severity::kINTERNAL_ERROR, stream.str().c_str());
|
|
}
|
|
#else
|
|
getLogger()->log(nvinfer1::ILogger::Severity::kINTERNAL_ERROR, stream.str().c_str());
|
|
#endif
|
|
PLUGIN_CUASSERT(cudaDeviceReset());
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
void TRTException::log(std::ostream& logStream) const
|
|
{
|
|
logStream << file << " (" << line << ") - " << name << " Error in " << function << ": " << status;
|
|
if (message != nullptr)
|
|
{
|
|
logStream << " (" << message << ")";
|
|
}
|
|
logStream << std::endl;
|
|
}
|
|
|
|
} // namespace nvinfer1::plugin
|