From 3045f29814856702e8c45600f751d6ed5a79139e Mon Sep 17 00:00:00 2001
From: project6-dev <dev@project6.ai>
Date: Thu, 13 Aug 2026 03:45:45 +0000
Subject: [PATCH] =?UTF-8?q?fix:=20enable=20corex=20topk=5Fsoftmax=20kernel?=
 =?UTF-8?q?=20=E2=80=94=20verified=20correct=20on=20real=20BI-V100?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

verify_topk_softmax.py results:
  IDs match: True (0/32 mismatches)
  Max weight diff: 0.00000003
  Speedup: 2.54x vs PyTorch (0.025ms vs 0.064ms)

Previous disable was based on speculation, not measurement.
---
 computility-run.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/computility-run.yaml b/computility-run.yaml
index c0e2a6e1..947b6a6b 100644
--- a/computility-run.yaml
+++ b/computility-run.yaml
@@ -46,6 +46,6 @@ env:
     - name: BI100_GDN_RESTORE_MODE
       value: hybrid64
     - name: BI100_MOE_COREX_TOPK_SOFTMAX
-      value: '0'
+      value: '1'
     - name: PYTORCH_CUDA_ALLOC_CONF
       value: expandable_segments:True