From 8b0538438ea9c2e3038d3da10090b20e08f46089 Mon Sep 17 00:00:00 2001
From: Chris Sidebottom <chris.sidebottom@arm.com>
Date: Thu, 31 Mar 2022 12:26:51 +0100
Subject: [PATCH] [CI] Run frontend tests for aarch64 in CI

Using `task_python_frontend_cpu.sh` to begin with to match those used
in `ci_cpu` - can add more frontends after this initial set is
functional.
---
 Jenkinsfile            | 23 +++++++++++++++++++++++
 jenkins/Jenkinsfile.j2 | 23 +++++++++++++++++++++++
 2 files changed, 46 insertions(+)

diff --git a/Jenkinsfile b/Jenkinsfile
index f685be9fdde6..f2fd8822462d 100755
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -851,6 +851,29 @@ stage('Test') {
       Utils.markStageSkippedForConditional('frontend: CPU')
     }
   },
+  'frontend: aarch64': {
+    if (!skip_ci && is_docs_only_build != 1) {
+      node('ARM') {
+        ws("workspace/exec_${env.EXECUTOR_NUMBER}/tvm/ut-python-arm") {
+          try {
+            init_git()
+            unpack_lib('arm', tvm_multilib)
+            timeout(time: max_time, unit: 'MINUTES') {
+              ci_setup(ci_arm)
+              sh (
+                script: "${docker_run} ${ci_arm} ./tests/scripts/task_python_frontend_cpu.sh",
+                label: 'Run Python frontend tests',
+              )
+            }
+          } finally {
+            junit 'build/pytest-results/*.xml'
+          }
+        }
+      }
+    } else {
+      Utils.markStageSkippedForConditional('frontend: aarch64')
+    }
+  },
   'docs: GPU': {
     if (!skip_ci) {
       node('TensorCore') {
diff --git a/jenkins/Jenkinsfile.j2 b/jenkins/Jenkinsfile.j2
index 016432e2c511..2b8eb646d350 100644
--- a/jenkins/Jenkinsfile.j2
+++ b/jenkins/Jenkinsfile.j2
@@ -848,6 +848,29 @@ stage('Test') {
       Utils.markStageSkippedForConditional('frontend: CPU')
     }
   },
+  'frontend: aarch64': {
+    if (!skip_ci && is_docs_only_build != 1) {
+      node('ARM') {
+        ws("workspace/exec_${env.EXECUTOR_NUMBER}/tvm/ut-python-arm") {
+          try {
+            init_git()
+            unpack_lib('arm', tvm_multilib)
+            timeout(time: max_time, unit: 'MINUTES') {
+              ci_setup(ci_arm)
+              sh (
+                script: "${docker_run} ${ci_arm} ./tests/scripts/task_python_frontend_cpu.sh",
+                label: 'Run Python frontend tests',
+              )
+            }
+          } finally {
+            junit 'build/pytest-results/*.xml'
+          }
+        }
+      }
+    } else {
+      Utils.markStageSkippedForConditional('frontend: aarch64')
+    }
+  },
   'docs: GPU': {
     if (!skip_ci) {
       node('TensorCore') {