Merge pull request #8 from bin913/add_run_op_cicd

Add run_op.sh in backend_test.yaml for cicd

Author: w1120029931-bit

.github/workflows/unittest.yaml

@@ -12,9 +12,12 @@ jobs:
   preprocess:
     runs-on: ubuntu-latest
     outputs:
-      labels: ${{ fromJSON(steps.get-labels.outputs.result) }}
+      changed_files: ${{ steps.changed-files.outputs.all_changed_files }}
       pr_id: ${{ steps.get-pr-id.outputs.PR_ID }}
+      labels: ${{ fromJSON(steps.get-labels.outputs.result) }}
     steps:
+      - id: changed-files
+        uses: tj-actions/changed-files@9426d40962ed5378910ee2e21d5f8c6fcbf2dd96  # v47.0.6
       - uses: actions/checkout@v4
 
       - id: wait-for-triage
@@ -75,3 +78,4 @@ jobs:
       runner_label: 'h100'
       gpu_check_script: 'tools/check_nvidia_gpu.sh'
       pr_id: ${{ needs.preprocess.outputs.pr_id }}
+      changed_files: ${{ join(needs.preprocess.outputs.changed_files, ' ') }}

conf/operators.yaml

@@ -54,7 +54,7 @@ ops:
     for:
       - sabs
     labels:
-      - abs
+      - sabs
     kind:
       - BLAS
     stages:
@@ -66,7 +66,7 @@ ops:
     for:
       - dabs
     labels:
-      - abs
+      - dabs
     kind:
       - BLAS
     stages:
@@ -78,7 +78,7 @@ ops:
     for:
       - cabs
     labels:
-      - abs
+      - cabs
     kind:
       - BLAS
     stages:

pyproject.toml

@@ -30,6 +30,7 @@ dependencies = [
 [project.optional-dependencies]
 test = [
     "pytest>=7.1.0",
+    "coverage==7.13.5",
     "numpy>=1.26",
     "scipy>=1.14",
     "cupy-cuda12x",

tests/conftest.py

@@ -36,16 +36,11 @@ def pytest_addoption(parser):
         help="device to run reference tests on",
     )
     parser.addoption(
-        (
-            "--mode"
-            if not (flag_blas.vendor_name == "kunlunxin" and torch.__version__ < "2.5")
-            else "--fg_mode"
-        ),  # TODO: fix pytest-* common --mode args,
-        action="store",
-        default="normal",
+        "--quick",
+        action="store_true",
+        default=False,
         required=False,
-        choices=["normal", "quick"],
-        help="run tests on normal or quick mode",
+        help="run tests on quick mode",
     )
     parser.addoption(
         "--record",
@@ -86,7 +81,7 @@ def pytest_configure(config):
     print(f"[correctness] reference backend: {ref_backend}", flush=True)
 
     global QUICK_MODE
-    QUICK_MODE = config.getoption("--mode") == "quick"
+    QUICK_MODE = config.getoption("--quick")
 
     global RECORD_LOG
     RECORD_LOG = config.getoption("--record") == "log"

tests/test_sgemm.py

@@ -203,7 +203,7 @@ def test_sgemm_empty(m, n, k):
 
 @pytest.mark.sgemm
 @pytest.mark.parametrize(
-    "alpha,beta", [(1.0, 0.0), (2.0, 0.0), (2.0, 0.5), (0.0, 1.0), (0.5, 1.5)]
+    "alpha,beta", [(1.0, 0.0), (2.0, 0.0), (2.0, 0.5), (0.0, 1.0), (0.5, 2.5)]
 )
 def test_sgemm_alpha_beta(alpha, beta):
     m, n, k = 256, 256, 256

tools/setup_vendor.sh

@@ -29,16 +29,15 @@ echo "Installing FlagBLAS for ${VENDOR} ..."
 case $VENDOR in
   nvidia)
     # Install PyTorch and Triton with CUDA support
-    uv pip install --index ${FLAGOS_PYPI} \
-        "torch==2.9.1+cu128" \
-        "torchvision==0.24.1+cu128" 
+    uv pip install torch==2.9.1 torchvision==0.24.1 torchaudio==2.9.1 \
+        --index-url https://download.pytorch.org/whl/cu128
     # Install FlagBLAS in editable mode
-    uv pip install -e .
-    uv pip install ".[test]"
 
     uv pip uninstall triton
-    uv pip install --index ${FLAGOS_PYPI} \
-        flagtree==0.5.0
+    RES="--index-url=https://resource.flagos.net/repository/flagos-pypi-hosted/simple"
+    python3.12 -m pip install flagtree===0.5.0 $RES
+    uv pip install -e .
+    uv pip install ".[test]"
     ;;
 
   iluvatar)