From e8cc450693c9792ad2e91bae45a9ceac47a586f4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E3=82=B3=E3=83=9E=E3=83=AA=E3=83=B3=E8=A6=AA=E8=A1=9B?=
 =?UTF-8?q?=E9=9A=8A?= <hello@tuna2134.dev>
Date: Tue, 1 Oct 2024 01:44:53 +0900
Subject: [PATCH] Update CI.yml

---
 .github/workflows/CI.yml | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
index ec1bf23..aea7117 100644
--- a/.github/workflows/CI.yml
+++ b/.github/workflows/CI.yml
@@ -130,3 +130,36 @@ jobs:
         with:
           command: upload
           args: --non-interactive --skip-existing wheels-*/*
+
+  push-docker:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      packages: write
+    strategy:
+      matrix:
+        tag: [cpu, cuda]
+        platform:
+        - linux/amd64
+        - linux/arm64
+    steps:
+      - uses: actions/checkout@v4
+      - name: Set up QEMU
+        uses: docker/setup-qemu-action@v3
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+      - name: Login to GitHub Container Registry
+        uses: docker/login-action@v3
+        with:
+          registry: ghcr.io
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+      - name: Build and push image
+        uses: docker/build-push-action@v6
+        with:
+          context: .
+          push: true
+          tags: |
+            ghcr.io/${{ github.repository }}:${{ matrix.tag }}
+          file: docker/${{ matrix.tag }}.Dockerfile
+          platforms: ${{ matrix.platform }}