From 9f144ae7bf7710399bab21899a208981d3d5c9e0 Mon Sep 17 00:00:00 2001
From: jagadeesh <jagadeeshj@ideas2it.com>
Date: Fri, 23 Jul 2021 20:02:33 +0530
Subject: [PATCH 1/2] feat: Add build script for kfserving torchserve image

---
 kubernetes/kfserving/README.md      | 20 ++++++++++++++------
 kubernetes/kfserving/build_image.sh | 29 +++++++++++++++++++++++++++++
 2 files changed, 43 insertions(+), 6 deletions(-)
 create mode 100755 kubernetes/kfserving/build_image.sh

diff --git a/kubernetes/kfserving/README.md b/kubernetes/kfserving/README.md
index 063ca62985..a66100c08a 100644
--- a/kubernetes/kfserving/README.md
+++ b/kubernetes/kfserving/README.md
@@ -6,21 +6,29 @@ Currently, KFServing supports the Inference API for all the existing models but
 
 ### Docker Image Building
 
-For CPU Image
+* To create a CPU based image
+
 ```
-docker build -t pytorch/torchserve-kfs:latest .
+./build_image.sh 
 ```
 
-For GPU Image
+* To create a CPU based image with custom tag
+
 ```
-docker build --build-arg BASE_IMAGE=pytorch/torchserve:latest-gpu -t pytorch/torchserve-kfs:latest-gpu .
+./build_image.sh -t <repository>/<image>:<tag>
 ```
 
-Push image to repository
+* To create a GPU based image
+
 ```
-docker push pytorch/torchserve-kfs:latest
+./build_image.sh -g 
 ```
 
+* To create a GPU based image with custom tag
+
+```
+./build_image.sh -g -t <repository>/<image>:<tag>
+```
 
 Individual Readmes for KFServing :
 
diff --git a/kubernetes/kfserving/build_image.sh b/kubernetes/kfserving/build_image.sh
new file mode 100755
index 0000000000..ea7b587327
--- /dev/null
+++ b/kubernetes/kfserving/build_image.sh
@@ -0,0 +1,29 @@
+#!/bin/bash
+
+DOCKER_TAG="pytorch/torchserve-kfs:latest"
+BASE_IMAGE="pytorch/torchserve:latest"
+
+for arg in "$@"
+do
+    case $arg in
+        -h|--help)
+          echo "options:"
+          echo "-h, --help  show brief help"
+          echo "-g, --gpu specify for gpu build"
+          echo "-t, --tag specify tag name for docker image"
+          exit 0
+          ;;
+        -g|--gpu)
+          DOCKER_TAG="pytorch/torchserve-kfs:latest-gpu"
+          BASE_IMAGE="pytorch/torchserve:latest-gpu"
+          shift
+          ;;
+        -t|--tag)
+          DOCKER_TAG="$2"
+          shift
+          shift
+          ;;
+    esac
+done
+
+DOCKER_BUILDKIT=1 docker build --file Dockerfile --build-arg BASE_IMAGE=$BASE_IMAGE -t "$DOCKER_TAG" .

From 13c7760eb9ae56dcff2ebdcde22201c1b0d8b0de Mon Sep 17 00:00:00 2001
From: jagadeesh <jagadeeshj@ideas2it.com>
Date: Mon, 26 Jul 2021 16:35:34 +0530
Subject: [PATCH 2/2] Trigger test