Rajeev Rao
87 lines
2.6 KiB
C++
87 lines
2.6 KiB
C++
/*
|
|
* SPDX-FileCopyrightText: Copyright (c) 1993-2022 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 TRT_PYTHON_FORWARD_DECLARATIONS_H
|
|
#define TRT_PYTHON_FORWARD_DECLARATIONS_H
|
|
|
|
// clang-format off
|
|
// Hack for missing declarations on Windows.
|
|
// These headers must be included before pybind11.h as some dependencies are missing.
|
|
#ifdef _MSC_VER
|
|
#include <cstdint>
|
|
using ssize_t = int64_t;
|
|
#endif // _MSC_VER
|
|
// clang-format on
|
|
|
|
#include <pybind11/pybind11.h>
|
|
|
|
#include "NvCaffeParser.h"
|
|
#include "NvInfer.h"
|
|
#include "NvInferPlugin.h"
|
|
#include "NvUffParser.h"
|
|
#include "onnx/NvOnnxParser.h"
|
|
|
|
// We need to avoid making copies of PluginField because it does not own any of it's members.
|
|
// When there are multiple PluginFields pointing to the same data in Python, bad things happen.
|
|
// Making this opaque allows us to create lists of PluginFields without creating unwanted copies.
|
|
PYBIND11_MAKE_OPAQUE(std::vector<nvinfer1::PluginField>);
|
|
|
|
namespace tensorrt
|
|
{
|
|
// Set some global namespace aliases.
|
|
namespace py = pybind11;
|
|
// This is for literal operators (like _a for default args)
|
|
using namespace pybind11::literals;
|
|
// Hack for situations where the C++ object does not own a member string/const char*.
|
|
// Cannot reference python strings, so we make a copy and keep it alive on the C++ side.
|
|
struct FallbackString
|
|
{
|
|
FallbackString() = default;
|
|
FallbackString(std::string other)
|
|
: mData{other}
|
|
{
|
|
}
|
|
FallbackString(py::str other)
|
|
: mData{std::string(other)}
|
|
{
|
|
}
|
|
const char* c_str() const
|
|
{
|
|
return mData.c_str();
|
|
}
|
|
const char* c_str()
|
|
{
|
|
return mData.c_str();
|
|
}
|
|
std::string mData{};
|
|
};
|
|
|
|
// Infer
|
|
void bindFoundationalTypes(py::module& m);
|
|
void bindPlugin(py::module& m);
|
|
void bindInt8(py::module& m);
|
|
void bindGraph(py::module& m);
|
|
void bindAlgorithm(py::module& m);
|
|
void bindCore(py::module& m);
|
|
// Parsers
|
|
void bindOnnx(py::module& m);
|
|
void bindUff(py::module& m);
|
|
void bindCaffe(py::module& m);
|
|
} // namespace tensorrt
|
|
|
|
#endif // TRT_PYTHON_FORWARD_DECLARATIONS_H
|