(tvm) branch unity_ci updated: upd

[syfeng]

This is an automated email from the ASF dual-hosted git repository.

syfeng pushed a commit to branch unity_ci
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/unity_ci by this push:
     new 79f95b7f85 upd
79f95b7f85 is described below

commit 79f95b7f85146f43ea819347a3107955710b8b28
Author: Siyuan Feng <hzfen...@sjtu.edu.cn>
AuthorDate: Mon Nov 27 20:14:14 2023 +0800

    upd
---
 ci/jenkins/unity_jenkinsfile.groovy | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/ci/jenkins/unity_jenkinsfile.groovy 
b/ci/jenkins/unity_jenkinsfile.groovy
index e32cc54a03..efd5ec12aa 100644
--- a/ci/jenkins/unity_jenkinsfile.groovy
+++ b/ci/jenkins/unity_jenkinsfile.groovy
@@ -246,6 +246,17 @@ def cpp_unittest(image) {
   )
 }
 
+def copy_build_artifacts(src, dst) {
+  sh(
+    script: """
+      if [ -f ${src} ]; then
+        cp ${src} ${dst}
+      fi
+    """,
+    label: 'Copy build artifacts',
+  )
+}
+
 stage('Build') {
   parallel 'CPU': {
       node('CPU') {
@@ -263,6 +274,8 @@ stage('Build') {
           init_git()
           sh "${docker_run} --no-gpu ${ci_gpu} 
./tests/scripts/task_config_build_gpu.sh build"
           make("${ci_gpu} --no-gpu")
+          
copy_build_artifacts('./build/3rdparty/libflash_attn/src/libflash_attn.so', 
'./build/libflash_attn.so')
+          
copy_build_artifacts('./build/3rdparty/cutlass_fpA_intB_gemm/cutlass_kernels/libfpA_intB_gemm.so',
 './build/libfpA_intB_gemm.so')
           pack_lib('gpu', tvm_lib_gpu)
         }
       }