ci(test): migrate OpenAI server tests to registered CI system (#16326)

Co-authored-by: Kangyan-Zhou <zky314343421@gmail.com>
This commit is contained in:
Alison Shao
2026-01-06 11:09:37 -08:00
committed by GitHub
parent 6e3fff1352
commit 0cbd8f3247
24 changed files with 86 additions and 35 deletions

View File

@@ -3,7 +3,6 @@ import tempfile
from contextlib import nullcontext
import torch
import torch.utils.cpp_extension
from packaging import version
from torch.cuda.memory import CUDAPluggableAllocator
@@ -90,6 +89,8 @@ def restore_symmetric_memory_context(saved_context):
def get_nccl_mem_pool():
global _allocator, _mem_pool, _cur_device
if _mem_pool is None:
import torch.utils.cpp_extension
out_dir = tempfile.gettempdir()
nccl_allocator_libname = "nccl_allocator"
torch.utils.cpp_extension.load_inline(