[Chore] Clean up pytorch helper functions in vllm.utils (#26908)

Signed-off-by: Isotr0py <mozf@mail2.sysu.edu.cn> Signed-off-by: isotr0py <2037008807@qq.com>
2025-10-19 00:48:22 +08:00
parent 5c2acb270a
commit 6ac5e06f7c
119 changed files with 772 additions and 714 deletions
--- a/tests/kernels/utils.py
+++ b/tests/kernels/utils.py
@@ -22,8 +22,8 @@ from vllm.utils import (
    STR_BACKEND_ENV_VAR,
    STR_FLASH_ATTN_VAL,
    STR_XFORMERS_ATTN_VAL,
-    make_tensor_with_pad,
 )
+from vllm.utils.torch_utils import make_tensor_with_pad

 # For now, disable "test_aot_dispatch_dynamic" since there are some
 # bugs related to this test in PyTorch 2.4.