Add support for aarch64

Change-Id: I5b80e63bde83ee0e203cdf4852bc44ec68059692 Signed-off-by: Henry Wang <henry.wang@arm.com> Jira: ENTOS-1324
kubeflow · Nov 1, 2019 · 2521555 · 2521555
1 parent de4847c
commit 2521555
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/py/kubeflow/tf_operator/util.py b/py/kubeflow/tf_operator/util.py
@@ -48,6 +48,8 @@ def run(command, cwd=None, env=None, dryrun=False):
   goarch = "amd64"
   if platform.machine() == "ppc64le":
     goarch = "ppc64le"
+  elif platform.machine() == "aarch64":
+    goarch = "arm64"
   extra_envs = {"GOOS": "linux", "GOARCH": goarch}
 
   if not env: