Add the build of adsbrain backend to build.py

build.py

@@ -538,6 +538,8 @@ def backend_repo(be):
         return 'tensorflow_backend'
     if be.startswith("openvino"):
         return 'openvino_backend'
+    if (be == 'adsbrain'):
+        return 'triton-adsbrain-backend'
     return '{}_backend'.format(be)
 
 
@@ -638,7 +640,11 @@ def pytorch_cmake_args(images):
 def onnxruntime_cmake_args(images, library_paths):
     cargs = [
         cmake_backend_arg('onnxruntime', 'TRITON_BUILD_ONNXRUNTIME_VERSION',
-                          None, TRITON_VERSION_MAP[FLAGS.version][2])
+                          None, TRITON_VERSION_MAP[FLAGS.version][2]),
+        cmake_backend_arg('onnxruntime', 'TRITON_BUILD_CUDA_HOME',
+                          None, "/usr/local/cuda"),
+        cmake_backend_arg('onnxruntime', 'TRITON_BUILD_CUDNN_HOME',
+                          None, "/usr/local/cuda"),
     ]
 
     # TRITON_ENABLE_GPU is already set for all backends in backend_cmake_args()
@@ -2273,6 +2279,7 @@ def enable_all():
             if (be in CORE_BACKENDS):
                 continue
 
+            github_organization = FLAGS.github_organization
             tagged_be_list = []
             if (be == 'openvino'):
                 tagged_be_list.append(
@@ -2283,6 +2290,8 @@ def enable_all():
                         if not skip:
                             tagged_be_list.append(tagged_backend(be, ver))
                         skip = False
+            if (be == 'adsbrain'):
+                github_organization = 'https://github.com/feihugis'
 
             # If armnn_tflite backend, source from external repo for git clone
             if be == 'armnn_tflite':

build_adsbrain.sh

@@ -0,0 +1,69 @@
+#!/bin/bash
+
+# Note cmake version 3.27.6 works; but cmake version 3.25 will not work.
+apt-get update
+apt-get install libssl-dev
+apt-get install rapidjson-dev
+apt-get install libboost-all-dev
+apt-get install libarchive-dev
+nv-hostengine -t
+distribution=$(. /etc/os-release;echo $ID$VERSION_ID | sed -e 's/\.//g')
+wget https://developer.download.nvidia.com/compute/cuda/repos/$distribution/x86_64/cuda-keyring_1.0-1_all.deb
+dpkg -i cuda-keyring_1.0-1_all.deb
+apt-get install -y datacenter-gpu-manager
+apt-get install libre2-dev
+apt-get install -y cl-base64
+apt-get install -y libb64-dev
+
+rm -rf checksum/
+rm -rf cmake_build
+rm -rf opt/
+rm -rf python
+rm -rf tritonserver/
+rm -rf onnxruntime
+
+# Build without container
+python build.py \
+  --enable-logging \
+  --enable-stats \
+  --enable-tracing \
+  --enable-metrics \
+  --enable-gpu-metrics \
+  --enable-gpu \
+  --endpoint=http \
+  --endpoint=adsbrain \
+  --repo-tag=common:r22.05 \
+  --repo-tag=core:r22.05 \
+  --repo-tag=backend:r22.05 \
+  --repo-tag=thirdparty:r22.05_ab \
+  --backend=adsbrain:r22.05_ab \
+  --backend=python:r22.05 \
+  --backend=onnxruntime:r22.05 \
+  --repoagent=checksum:r22.05 \
+  --build-type=Release \
+  --build-parallel=80 \
+  --no-container-build \
+  --build-dir=`pwd` \
+  --verbose
+
+# Build with container
+python build.py \
+  --enable-logging \
+  --enable-stats \
+  --enable-tracing \
+  --enable-metrics \
+  --enable-gpu-metrics \
+  --enable-gpu \
+  --endpoint=http \
+  --endpoint=adsbrain \
+  --repo-tag=common:r22.05 \
+  --repo-tag=core:r22.05 \
+  --repo-tag=backend:r22.05 \
+  --repo-tag=thirdparty:r22.05_ab \
+  --backend=adsbrain:r22.05_ab \
+  --backend=python:r22.05 \
+  --backend=onnxruntime \
+  --repoagent=checksum:r22.05 \
+  --build-type=Release \
+  --build-parallel=80 \
+  --verbose