From 3331944547c8d4c758f88c9612294ec18a7c57bc Mon Sep 17 00:00:00 2001 From: wozeparrot Date: Tue, 4 Aug 2026 02:16:44 +0800 Subject: [PATCH 1/2] gptoss: fix moe routing (#17377) --- extra/gemm/moe_routing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extra/gemm/moe_routing.py b/extra/gemm/moe_routing.py index 80b55b42c4..e683b83067 100644 --- a/extra/gemm/moe_routing.py +++ b/extra/gemm/moe_routing.py @@ -5,7 +5,7 @@ BLOCK_ROW = 256 def _sharded_invalids(shape:tuple[int, ...], dtype, device) -> Tensor: if isinstance(device, tuple): - return Tensor(Tensor.invalids(shape[0] // len(device), *shape[1:], dtype=dtype, device=device).uop.multi(0), device=device) + return Tensor.invalids(*shape, dtype=dtype, device=device[0]).shard(device, axis=0) return Tensor.invalids(*shape, dtype=dtype, device=device) def _atomic_add(device:str) -> str: From a2385ae21d8b27f264ebaf95bdfaae49d7474e2b Mon Sep 17 00:00:00 2001 From: chenyu Date: Mon, 3 Aug 2026 15:37:58 -0400 Subject: [PATCH 2/2] MAX_LINE_COUNT=26000 (#17378) oh well --- .github/workflows/test.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index cb1cddf171..4d272ca048 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -219,8 +219,8 @@ jobs: run: python3 test/external/external_benchmark_schedule.py - name: Run process replay tests uses: ./.github/actions/process-replay - - name: Repo line count < 25000 lines - run: MAX_LINE_COUNT=25000 python sz.py + - name: Repo line count <= 26000 lines + run: MAX_LINE_COUNT=26000 python sz.py spec: strategy: