update torch2.5

CMakeLists.txt

@@ -51,7 +51,7 @@ set(HIP_SUPPORTED_ARCHS "gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx
 # versions are derived from docker/Dockerfile.rocm
 #
 set(TORCH_SUPPORTED_VERSION_CUDA "2.7.0")
-set(TORCH_SUPPORTED_VERSION_ROCM "2.4.1")
+set(TORCH_SUPPORTED_VERSION_ROCM "2.5.1")
 
 #
 # Try to find python package with an executable that exactly matches

requirements/rocm.txt

@@ -20,7 +20,7 @@ runai-model-streamer-s3==0.11.0
 setuptools_scm>=8
 cmake==3.29
 
-torch == 2.4.1
+torch == 2.5.1
 triton == 3.0.0
 flash_attn == 2.6.1
 flash_mla == 1.0.0

setup.py

@@ -558,10 +558,10 @@ def get_version_add(sha: Optional[str] = None) -> str:
         if sha != 'Unknown':
             if sha is None:
                 sha = get_sha(vllm_root)
-            if (major, minor) == ('2', '4'):
+            if (major, minor) >= ('2', '5'):
                 version = 'das.opt1.alpha.' + sha[:7]
     else:
-        if (major, minor) == ('2', '4'):
+        if (major, minor) >= ('2', '5'):
             version = 'das.opt1.alpha'