mirror of
https://github.com/lingble/meta-tegra.git
synced 2026-01-13 00:45:13 +00:00
62 lines
2.1 KiB
BlitzBasic
62 lines
2.1 KiB
BlitzBasic
SUMMARY = "Python bindings for TensorRT"
|
|
HOMEPAGE = "http://developer.nvidia.com/tensorrt"
|
|
LICENSE = "Proprietary"
|
|
LIC_FILES_CHKSUM = "file://python/packaging/LICENSE.txt;md5=c291e0a531e08d4914e269730ba2f70d"
|
|
|
|
DEPENDS = "python3-pybind11 tensorrt-core tensorrt-plugins"
|
|
|
|
COMPATIBLE_MACHINE = "(tegra)"
|
|
|
|
inherit setuptools3 cmake cuda
|
|
|
|
SRC_REPO = "github.com/NVIDIA/TensorRT.git;protocol=https"
|
|
SRCBRANCH = "release/8.4"
|
|
SRC_URI = "git://${SRC_REPO};branch=${SRCBRANCH} \
|
|
file://0001-Fixups-for-cross-building-in-OE.patch \
|
|
"
|
|
# 8.4.1 tag
|
|
SRCREV = "b55c4710ce01f076c26710a48879fcb2661be4a9"
|
|
|
|
S = "${WORKDIR}/git"
|
|
|
|
OECMAKE_SOURCEPATH = "${S}/python"
|
|
SETUPTOOLS_SETUP_PATH = "${B}"
|
|
|
|
EXTRA_OECMAKE = "-DONNX_INC_DIR=${STAGING_INCDIR} -DPYBIND11_DIR=${STAGING_DIR_TARGET} \
|
|
-DTARGET=${HOST_ARCH} -DCMAKE_BUILD_TYPE=Release \
|
|
-DPY_INCLUDE=${STAGING_INCDIR}/${PYTHON_DIR} -DEXT_PATH=${STAGING_INCDIR}"
|
|
|
|
CXXFLAGS += "${CUDA_CXXFLAGS}"
|
|
|
|
do_configure() {
|
|
cmake_do_configure
|
|
TRT_MAJOR=$(awk '/^#define NV_TENSORRT_MAJOR/ {print $3}' ${STAGING_INCDIR}/NvInferVersion.h)
|
|
TRT_MINOR=$(awk '/^#define NV_TENSORRT_MINOR/ {print $3}' ${STAGING_INCDIR}/NvInferVersion.h)
|
|
TRT_PATCH=$(awk '/^#define NV_TENSORRT_PATCH/ {print $3}' ${STAGING_INCDIR}/NvInferVersion.h)
|
|
TRT_BUILD=$(awk '/^#define NV_TENSORRT_BUILD/ {print $3}' ${STAGING_INCDIR}/NvInferVersion.h)
|
|
TRT_VERSION=${TRT_MAJOR}.${TRT_MINOR}.${TRT_PATCH}.${TRT_BUILD}
|
|
TRT_MAJMINPATCH=${TRT_MAJOR}.${TRT_MINOR}.${TRT_PATCH}
|
|
varsubst() {
|
|
sed -e "s|\#\#TENSORRT_VERSION\#\#|${TRT_VERSION}|g" \
|
|
-e "s|\#\#TENSORRT_MAJMINPATCH\#\#|${TRT_MAJMINPATCH}|g" $1 >$2
|
|
}
|
|
|
|
rm -rf ${B}/tensorrt
|
|
mkdir ${B}/tensorrt
|
|
varsubst ${S}/python/packaging/setup.cfg ${B}/setup.cfg
|
|
varsubst ${S}/python/packaging/setup.py ${B}/setup.py
|
|
varsubst ${S}/python/packaging/tensorrt/__init__.py ${B}/tensorrt/__init__.py
|
|
cp ${S}/python/packaging/LICENSE.txt ${B}/
|
|
}
|
|
|
|
do_compile() {
|
|
cmake_do_compile
|
|
setuptools3_do_compile
|
|
}
|
|
|
|
do_install() {
|
|
setuptools3_do_install
|
|
}
|
|
|
|
RDEPENDS:${PN} = "python3-ctypes python3-numpy"
|