diff --git a/computility-run.yaml b/computility-run.yaml index b009c3fb..7d356832 100644 --- a/computility-run.yaml +++ b/computility-run.yaml @@ -49,4 +49,6 @@ env: value: '1' - name: PYTORCH_CUDA_ALLOC_CONF value: max_split_size_mb:512 + - name: LD_PRELOAD + value: /workspace/qwen3_6_scripts/cccl_preload/libcccl_allocator.so