From 42cf124f20b51830375f369be0b14889c1e8a802 Mon Sep 17 00:00:00 2001
From: tsai <jackalcooper@gmail.com>
Date: Thu, 10 Feb 2022 12:09:07 +0800
Subject: [PATCH 1/2] check in files

---
 .github/workflows/build.yml | 4 ++++
 docker/Dockerfile           | 6 +++++-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 895f7362b..486bfbbd6 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -42,14 +42,17 @@ jobs:
             policy: "manylinux2014"
             platform: "x86_64"
             CUDA_BASE_IMAGE: "nvidia/cuda:11.3.1-cudnn8-devel-centos7"
+            target: "manylinux-with-trt"
           - tag-suffix: "cuda11.2"
             policy: "manylinux2014"
             platform: "x86_64"
             CUDA_BASE_IMAGE: "nvidia/cuda:11.2.2-cudnn8-devel-centos7"
+            target: "manylinux"
           - tag-suffix: "cuda10.2"
             policy: "manylinux2014"
             platform: "x86_64"
             CUDA_BASE_IMAGE: "nvidia/cuda:10.2-cudnn7-devel-centos7"
+            target: "manylinux"
           - tag-suffix: "cpu"
             policy: "manylinux2014"
             platform: "x86_64"
@@ -102,6 +105,7 @@ jobs:
           cache-from: type=registry,ref=${{ env.DOCKER_HUB_NAMESPACE }}/${{ env.DOCKER_REPO }}:latest
           cache-to: type=inline
           context: ./docker/
+          target: ${ matrix.target }
           build-args: |
             POLICY
             PLATFORM
diff --git a/docker/Dockerfile b/docker/Dockerfile
index e2f0bc3ee..a7e1545d5 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -149,7 +149,7 @@ COPY --from=build_cpython310 /opt/_internal /opt/_internal/
 RUN manylinux-entrypoint /build_scripts/finalize-python.sh
 
 
-FROM runtime_base
+FROM runtime_base AS manylinux
 COPY --from=build_git /manylinux-rootfs /
 COPY --from=build_swig /manylinux-rootfs /
 COPY --from=build_cpython /manylinux-rootfs /
@@ -174,4 +174,8 @@ RUN curl -L $BAZEL_URL -o /usr/local/bin/bazel \
     && chmod +x /usr/local/bin/bazel
 RUN yum install -y wget nasm rdma-core-devel rsync gdb ninja-build openblas-static devtoolset-7-gcc* vim ccache htop
 
+FROM manylinux AS manylinux-with-trt
+ARG trt=...
+RUN install trt ${}
+
 CMD ["/bin/bash"]

From fb6d11383d51e21318364601d331cc5fca7a88f8 Mon Sep 17 00:00:00 2001
From: tsai <jackalcooper@gmail.com>
Date: Fri, 25 Feb 2022 22:24:56 +0800
Subject: [PATCH 2/2] refine

---
 .github/workflows/build.yml | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 486bfbbd6..1fdc37efa 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -48,6 +48,11 @@ jobs:
             platform: "x86_64"
             CUDA_BASE_IMAGE: "nvidia/cuda:11.2.2-cudnn8-devel-centos7"
             target: "manylinux"
+          - tag-suffix: "cuda11.0"
+            policy: "manylinux2014"
+            platform: "x86_64"
+            CUDA_BASE_IMAGE: "nvidia/cuda:11.0.3-cudnn8-devel-centos7"
+            target: "manylinux"
           - tag-suffix: "cuda10.2"
             policy: "manylinux2014"
             platform: "x86_64"