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 84fdb9a364 upd
84fdb9a364 is described below
commit 84fdb9a364a0b1c09d3c11976961a4908621d6b6
Author: Siyuan Feng <[email protected]>
AuthorDate: Mon Nov 27 21:34:24 2023 +0800
upd
---
ci/jenkins/unity_jenkinsfile.groovy | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/ci/jenkins/unity_jenkinsfile.groovy
b/ci/jenkins/unity_jenkinsfile.groovy
index efd5ec12aa..31beef44df 100644
--- a/ci/jenkins/unity_jenkinsfile.groovy
+++ b/ci/jenkins/unity_jenkinsfile.groovy
@@ -41,6 +41,8 @@ ci_hexagon =
'tlcpackstaging/ci_hexagon:20230504-142417-4d37a0a0'
tvm_lib = 'build/libtvm.so, build/libtvm_runtime.so, build/config.cmake'
tvm_lib_gpu = tvm_lib + ", build/libfpA_intB_gemm.so, build/libflash_attn.so"
+tvm_ext_lib_gpu = "build/3rdparty/libflash_attn/src/libflash_attn.so,
build/3rdparty/cutlass_fpA_intB_gemm/cutlass_kernels/libfpA_intB_gemm.so"
+tvm_lib_gpu = tvm_lib_gpu + ", " + tvm_ext_lib_gpu
def pack_lib(name, libs) {
sh """
@@ -274,8 +276,6 @@ 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)
}
}
