feat(triton-linalg): modify ci script

Cambricon · Jul 5, 2024 · 6e7e4b1 · 6e7e4b1
1 parent 162315b
commit 6e7e4b1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/ci_script/file_guard.py b/.github/ci_script/file_guard.py
@@ -1,8 +1,9 @@
 import time
 import sys
 import os
+import argparse
 
-def file_guard():
+def file_guard(guard_status_file, guard_log_file):
     # where stores the last position that pointer pointed to.
     where= 0
     while True: