feat(triton-linalg): modify ci pepeline

Cambricon · Jul 5, 2024 · 50e7097 · 50e7097
1 parent 427c607
commit 50e7097
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 13 deletions.
diff --git a/.github/ci_script/triton-linalg-ci_script.sh b/.github/ci_script/triton-linalg-ci_script.sh
@@ -44,7 +44,7 @@ echo "working" > "$request_root/status"
 chmod o+w "$request_root/status"
 
 if [ ! -f  "$request_root/log" ];then
-	touch "$request_root/log"
+    touch "$request_root/log"
 fi
 
 chmod o+w "$request_root/log"

diff --git a/tools/ci/daily/triton-linalg_daliy.pipeline b/tools/ci/daily/triton-linalg_daliy.pipeline
@@ -2,19 +2,22 @@ library "cambricon-pipe-lib@master"
 cnpipe {
     task('clone') {
 	stage 'clone'
-        runOnCloud false
         node {
-             label "compiler_test_10.100.146.123"
+            labelSelector "cambricon.com/mm-daily":true
+            cardType 'MLU370'
         }
         container {
             networkPolicy "cncl-no-internnet-access"
             image 'yellow.hub.cambricon.com/genesis/devel/x86_64/triton_linalg:1.0.0-x86_64-ubuntu2004-prebuild-thirdparty-py_3_10'
             runArgs "--network=host --privileged -v /usr/bin/cnmon:/usr/bin/cnmon --device=/dev/cambricon_dev0:/dev/cambricon_dev0 --device=/dev/cambricon_ctl"
         }
         resReq {
-            reqMemory '40Gi'
             reqMlus 1
             lmtMlus 1
+            reqCpu 30
+            lmtCpu 30
+            reqMemory '40Gi'
+            lmtMemory '40Gi'
         }
         stage 'clone'
         script '''
@@ -32,19 +35,22 @@ cnpipe {
     }
     task('check_pr') {
         stage 'check_pr'
-        runOnCloud false
         node {
-            label "compiler_test_10.100.146.123"
+            labelSelector "cambricon.com/mm-daily":true
+            cardType 'MLU370'
         }
         container {
             networkPolicy "cncl-no-internnet-access"
             image 'yellow.hub.cambricon.com/genesis/devel/x86_64/triton_linalg:1.0.0-x86_64-ubuntu2004-prebuild-thirdparty-py_3_10'
             runArgs "--network=host --privileged -v /usr/bin/cnmon:/usr/bin/cnmon --device=/dev/cambricon_dev0:/dev/cambricon_dev0 --device=/dev/cambricon_ctl"
         }
         resReq {
-            reqMemory '40Gi'
             reqMlus 1
             lmtMlus 1
+            reqCpu 30
+            lmtCpu 30
+            reqMemory '40Gi'
+            lmtMemory '40Gi'
         }
         unstash 'triton-linalg-pr'
         script '''
@@ -60,19 +66,22 @@ cnpipe {
     }
     task('build') {
         stage 'build'
-        runOnCloud false
         node {
-            label "compiler_test_10.100.146.123"
+            labelSelector "cambricon.com/mm-daily":true
+            cardType 'MLU370'
         }
         container {
             networkPolicy "cncl-no-internnet-access"
             image 'yellow.hub.cambricon.com/genesis/devel/x86_64/triton_linalg:1.0.0-x86_64-ubuntu2004-prebuild-thirdparty-py_3_10'
             runArgs "--network=host --privileged -v /usr/bin/cnmon:/usr/bin/cnmon --device=/dev/cambricon_dev0:/dev/cambricon_dev0 --device=/dev/cambricon_ctl"
         }
         resReq {
-            reqMemory '40Gi'
             reqMlus 1
             lmtMlus 1
+            reqCpu 30
+            lmtCpu 30
+            reqMemory '40Gi'
+            lmtMemory '40Gi'
         }
         unstash 'triton-linalg-pr'
         script '''
@@ -89,19 +98,22 @@ cnpipe {
     }
     task('test') {
         stage 'test'
-        runOnCloud false
         node {
-            label "compiler_test_10.100.146.123"
+            labelSelector "cambricon.com/mm-daily":true
+            cardType 'MLU370'
         }
         container {
             networkPolicy "cncl-no-internnet-access"
             image 'yellow.hub.cambricon.com/genesis/devel/x86_64/triton_linalg:1.0.0-x86_64-ubuntu2004-prebuild-thirdparty-py_3_10'
             runArgs "--network=host --privileged -v /usr/bin/cnmon:/usr/bin/cnmon --device=/dev/cambricon_dev0:/dev/cambricon_dev0 --device=/dev/cambricon_ctl"
         }
         resReq {
-            reqMemory '40Gi'
             reqMlus 1
             lmtMlus 1
+            reqCpu 30
+            lmtCpu 30
+            reqMemory '40Gi'
+            lmtMemory '40Gi'
         }
         unstash 'triton-linalg-build'
         script '''