diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 53f2bd8bd..f3726b7b2 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -40,6 +40,15 @@ jobs: ghcr.io/${{ steps.setup.outputs.lowercase-repo }}:${{ github.ref_name }}-${{ env.SHORT_SHA }}-amd64 cache-from: type=gha cache-to: type=gha,mode=max + - name: Build and push TensorRT (x86 GPU) + uses: docker/bake-action@v3 + with: + push: true + targets: tensorrt + files: docker/tensorrt/trt.hcl + set: | + tensorrt.tags=ghcr.io/${{ steps.setup.outputs.lowercase-repo }}:${{ github.ref_name }}-${{ env.SHORT_SHA }}-tensorrt + *.cache-from=type=gha - name: Build and push arm64 standard build uses: docker/build-push-action@v4 with: @@ -61,15 +70,6 @@ jobs: set: | rpi.tags=ghcr.io/${{ steps.setup.outputs.lowercase-repo }}:${{ github.ref_name }}-${{ env.SHORT_SHA }}-rpi *.cache-from=type=gha - - name: Build and push TensorRT (x86 GPU) - uses: docker/bake-action@v3 - with: - push: true - targets: tensorrt - files: docker/tensorrt/trt.hcl - set: | - tensorrt.tags=ghcr.io/${{ steps.setup.outputs.lowercase-repo }}:${{ github.ref_name }}-${{ env.SHORT_SHA }}-tensorrt - *.cache-from=type=gha - name: Build and push TensorRT (Jetson, Jetpack 4) env: ARCH: arm64