[Main] Upgrade PTA to 2.9.0 (#6112)
### What this PR does / why we need it?
Upgrade PTA to 2.9.0
- vLLM version: v0.13.0
- vLLM main:
d68209402d
---------
Signed-off-by: wjunLu <wjunlu217@gmail.com>
This commit is contained in:
@@ -22,9 +22,9 @@ find_package(Torch REQUIRED)
|
||||
|
||||
run_python(TORCH_VERSION
|
||||
"import torch; print(torch.__version__)" "Failed to locate torch path")
|
||||
# check torch version is 2.8.0
|
||||
if(NOT ${TORCH_VERSION} VERSION_EQUAL "2.8.0")
|
||||
message(FATAL_ERROR "Expected PyTorch version 2.8.0, but found ${TORCH_VERSION}")
|
||||
# check torch version is 2.9.0
|
||||
if(NOT ${TORCH_VERSION} VERSION_EQUAL "2.9.0")
|
||||
message(FATAL_ERROR "Expected PyTorch version 2.9.0, but found ${TORCH_VERSION}")
|
||||
endif()
|
||||
|
||||
set(RUN_MODE "npu" CACHE STRING "cpu/sim/npu")
|
||||
|
||||
Reference in New Issue
Block a user