Skip to content

Commit 57d2cd4

Browse files
larryliu0820malfet
authored andcommitted
Update aoti.cmake to enforce torch version (#696)
As titled. This is to make sure we are locating the libtorch.so/dylib with the correct version number.
1 parent 8683ca1 commit 57d2cd4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

runner/aoti.cmake

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ ENDIF()
88

99
find_package(CUDA)
1010

11-
find_package(Torch)
11+
find_package(Torch 2.4.0)
1212
if(Torch_FOUND)
1313
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -g ${TORCH_CXX_FLAGS} -fpermissive")
1414

0 commit comments

Comments
 (0)