Skip to content

GitLab

Commit 675d3e87 authored by zhuwenwen's avatar zhuwenwen
Browse files

update torch2.5

parent 5f71c3f8
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
CMakeLists.txt
View file @ 675d3e87
...@@ -51,7 +51,7 @@ set(HIP_SUPPORTED_ARCHS "gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx ...@@ -51,7 +51,7 @@ set(HIP_SUPPORTED_ARCHS "gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx
# versions are derived from docker/Dockerfile.rocm # versions are derived from docker/Dockerfile.rocm
# #
set(TORCH_SUPPORTED_VERSION_CUDA "2.7.0") 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 # Try to find python package with an executable that exactly matches
......
requirements/rocm.txt
View file @ 675d3e87
...@@ -20,7 +20,7 @@ runai-model-streamer-s3==0.11.0 ...@@ -20,7 +20,7 @@ runai-model-streamer-s3==0.11.0
setuptools_scm>=8 setuptools_scm>=8
cmake==3.29 cmake==3.29
torch == 2.4.1 torch == 2.5.1
triton == 3.0.0 triton == 3.0.0
flash_attn == 2.6.1 flash_attn == 2.6.1
flash_mla == 1.0.0 flash_mla == 1.0.0
......
setup.py
View file @ 675d3e87
...@@ -558,10 +558,10 @@ def get_version_add(sha: Optional[str] = None) -> str: ...@@ -558,10 +558,10 @@ def get_version_add(sha: Optional[str] = None) -> str:
if sha != 'Unknown': if sha != 'Unknown':
if sha is None: if sha is None:
sha = get_sha(vllm_root) sha = get_sha(vllm_root)
if (major, minor) == ('2', '4'): if (major, minor) >= ('2', '5'):
version = 'das.opt1.alpha.' + sha[:7] version = 'das.opt1.alpha.' + sha[:7]
else: else:
if (major, minor) == ('2', '4'): if (major, minor) >= ('2', '5'):
version = 'das.opt1.alpha' version = 'das.opt1.alpha'
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment