Skip to content

Commit e5b34e2

Browse files
committed
Updating name of the variabl to presue it's purpose
1 parent 07d64fe commit e5b34e2

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

CMakeLists.txt

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -206,7 +206,7 @@ else()
206206
"libmkl_vml_def.so.1"
207207
)
208208
endif()
209-
set(OPENCV_LIBS
209+
set(TORCHVISION_LIBS
210210
$<IF:$<BOOL:${RHEL_BUILD}>,libjpeg.so.62,libjpeg.so>
211211
$<IF:$<BOOL:${RHEL_BUILD}>,libpng16.so.16,libpng16.so>
212212
)
@@ -222,7 +222,7 @@ if (${TRITON_PYTORCH_DOCKER_BUILD})
222222
OUTPUT
223223
${PT_LIBS}
224224
${LIBTORCH_LIBS}
225-
${OPENCV_LIBS}
225+
${TORCHVISION_LIBS}
226226
LICENSE.pytorch
227227
include/torch
228228
include/torchvision
@@ -264,7 +264,7 @@ if (${TRITON_PYTORCH_DOCKER_BUILD})
264264
COMMENT "Extracting pytorch and torchvision libraries and includes from ${TRITON_PYTORCH_DOCKER_IMAGE}"
265265
VERBATIM
266266
)
267-
add_custom_target(ptlib_target DEPENDS ${PT_LIBS} ${LIBTORCH_LIBS} ${OPENCV_LIBS})
267+
add_custom_target(ptlib_target DEPENDS ${PT_LIBS} ${LIBTORCH_LIBS} ${TORCHVISION_LIBS})
268268
add_library(ptlib SHARED IMPORTED GLOBAL)
269269
add_dependencies(ptlib ptlib_target)
270270

@@ -430,7 +430,7 @@ install(
430430

431431
if (${TRITON_PYTORCH_DOCKER_BUILD})
432432
set(PT_LIB_PATHS "")
433-
FOREACH(plib ${PT_LIBS} ${LIBTORCH_LIBS} ${OPENCV_LIBS})
433+
FOREACH(plib ${PT_LIBS} ${LIBTORCH_LIBS} ${TORCHVISION_LIBS})
434434
set(PT_LIB_PATHS ${PT_LIB_PATHS} "${CMAKE_CURRENT_BINARY_DIR}/${plib}")
435435
ENDFOREACH(plib)
436436

@@ -450,7 +450,7 @@ if (${TRITON_PYTORCH_DOCKER_BUILD})
450450
)
451451
endif() # TRITON_PYTORCH_ENABLE_TORCHTRT
452452

453-
FOREACH(plib ${PT_LIBS} ${LIBTORCH_LIBS} ${OPENCV_LIBS})
453+
FOREACH(plib ${PT_LIBS} ${LIBTORCH_LIBS} ${TORCHVISION_LIBS})
454454
install(
455455
CODE
456456
"EXECUTE_PROCESS(

0 commit comments

Comments
 (0)