From 2c1eb4fc850bed06b6a2f4e4d32f1678b02c26b0 Mon Sep 17 00:00:00 2001 From: Vincent Li Date: Thu, 11 Jun 2026 07:56:38 +0000 Subject: [PATCH 1/2] updates --- .github/ci_config.yaml | 10 +++++----- .github/workflows/ci_test.yml | 7 +++++++ 2 files changed, 12 insertions(+), 5 deletions(-) diff --git a/.github/ci_config.yaml b/.github/ci_config.yaml index 99eab3c0..015bf33b 100644 --- a/.github/ci_config.yaml +++ b/.github/ci_config.yaml @@ -14,7 +14,7 @@ platforms: CUDA_ARCH: sm_80,sm_86,sm_89,sm_90 APT_MIRROR: https://mirrors.tuna.tsinghua.edu.cn/ubuntu PIP_INDEX_URL: https://pypi.org/simple - InfiniCore_BRANCH: main + InfiniCore_BRANCH: __Branch_Name__ docker_args: - "--user=root" - "--network=host" @@ -80,7 +80,7 @@ platforms: BASE_IMAGE: cr.metax-tech.com/public-ai-release/maca/vllm-metax:0.10.2-maca.ai3.2.1.7-torch2.6-py310-ubuntu22.04-amd64 APT_MIRROR: https://mirrors.tuna.tsinghua.edu.cn/ubuntu PIP_INDEX_URL: https://pypi.org/simple - InfiniCore_BRANCH: main + InfiniCore_BRANCH: __Branch_Name__ docker_args: - "--user=root" - "--network=host" @@ -118,7 +118,7 @@ platforms: BASE_IMAGE: sh-harbor.mthreads.com/mcctest/vllm_musa:20251112_hygon APT_MIRROR: https://mirrors.tuna.tsinghua.edu.cn/ubuntu PIP_INDEX_URL: https://pypi.org/simple - InfiniCore_BRANCH: main + InfiniCore_BRANCH: __Branch_Name__ docker_args: - "--user=root" - "--network=host" @@ -186,7 +186,7 @@ platforms: BASE_IMAGE: pepe_working:latest APT_MIRROR: https://mirrors.tuna.tsinghua.edu.cn/ubuntu PIP_INDEX_URL: https://pypi.org/simple - InfiniCore_BRANCH: main + InfiniCore_BRANCH: __Branch_Name__ docker_args: - "--user=root" - "--network=host" @@ -285,7 +285,7 @@ platforms: BASE_IMAGE: image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.4.1-ubuntu22.04-dtk25.04.1-py3.10 APT_MIRROR: https://mirrors.tuna.tsinghua.edu.cn/ubuntu PIP_INDEX_URL: https://pypi.org/simple - InfiniCore_BRANCH: main + InfiniCore_BRANCH: __Branch_Name__ docker_args: - "--user=root" - "--network=host" diff --git a/.github/workflows/ci_test.yml b/.github/workflows/ci_test.yml index e07fee9a..d5ae9fff 100644 --- a/.github/workflows/ci_test.yml +++ b/.github/workflows/ci_test.yml @@ -5,6 +5,12 @@ on: branches: ["main"] types: [opened, reopened] # 只在 PR 创建/重新打开时自动跑 workflow_dispatch: # 支持在 Actions 页手动重跑 + inputs: + infinicore_branch: + description: "InfiniCore Branch Name" + required: false + default: "main" + type: string jobs: ci: @@ -12,4 +18,5 @@ jobs: with: config_path: .github/ci_config.yaml ci_ref: infiniCore_ci + infinicore_branch: ${{ github.event.inputs.infinicore_branch || 'main' }} secrets: inherit From acf7c97fe127213eb6ede0c8db71ad0002060a2a Mon Sep 17 00:00:00 2001 From: Vincent Li Date: Fri, 12 Jun 2026 05:31:03 +0000 Subject: [PATCH 2/2] update --- examples/bench.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/examples/bench.py b/examples/bench.py index 41cb54ac..018f7a8a 100644 --- a/examples/bench.py +++ b/examples/bench.py @@ -146,7 +146,12 @@ def get_test_cases( return case_dict -with open("examples/bench_prompt.md", "r") as f: +prompt_path = ( + "examples/bench_prompt.md" + if os.path.isfile("examples/bench_prompt.md") + else "InfiniLM/examples/bench_prompt.md" +) +with open(prompt_path, "r") as f: prompt = f.read()