From daa8067080c1a82bcd919d796e2397c6a477f61c Mon Sep 17 00:00:00 2001
From: Claude <claude@anthropic.com>
Date: Thu, 13 Aug 2026 11:22:32 +0000
Subject: [PATCH] fix: remove expandable_segments:True from
 PYTORCH_CUDA_ALLOC_CONF

CoreX CUDACachingAllocator does not support expandable_segments.
Setting it causes ASSERT failure at startup:
  'expandable_segments:True not supported on corex CUDACachingAllocator'

Replace with max_split_size_mb:512 only.
---
 computility-run.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/computility-run.yaml b/computility-run.yaml
index 015fca7f..a343a690 100644
--- a/computility-run.yaml
+++ b/computility-run.yaml
@@ -48,6 +48,6 @@ env:
     - name: BI100_MOE_COREX_TOPK_SOFTMAX
       value: '1'
     - name: PYTORCH_CUDA_ALLOC_CONF
-      value: expandable_segments:True
+      value: max_split_size_mb:512
     - name: LD_PRELOAD
       value: /workspace/qwen3_6_scripts/cccl_preload/libcccl_allocator.so