@@ -165,7 +165,7 @@ set(PT_LIBS
165165if (${TRITON_PYTORCH_ENABLE_TORCHVISION} )
166166 set (PT_LIBS
167167 ${PT_LIBS}
168- $<IF:$<BOOL :RHEL_BUILD>,libtorchvision.so,libtorchvision.so.1>
168+ $<IF:$<BOOL :${ RHEL_BUILD} >,libtorchvision.so,libtorchvision.so.1>
169169 )
170170endif () # TRITON_PYTORCH_ENABLE_TORCHVISION
171171
@@ -216,8 +216,8 @@ set(OPENCV_LIBS
216216 "libopencv_calib3d.so"
217217 "libopencv_flann.so"
218218 "libopencv_features2d.so"
219- $<IF:$<BOOL :RHEL_BUILD>,libjpeg.so.62,libjpeg.so>
220- $<IF:$<BOOL :RHEL_BUILD>,libpng16.so.16,libpng16.so>
219+ $<IF:$<BOOL :${ RHEL_BUILD} >,libjpeg.so.62,libjpeg.so>
220+ $<IF:$<BOOL :${ RHEL_BUILD} >,libpng16.so.16,libpng16.so>
221221)
222222
223223# The patchelf commands ensure the MKL libraries are loaded correctly during runtime
@@ -383,7 +383,7 @@ if (${TRITON_PYTORCH_DOCKER_BUILD})
383383 if (${TRITON_PYTORCH_ENABLE_TORCHVISION} )
384384 set (TRITON_PYTORCH_LIBS
385385 ${TRITON_PYTORCH_LIBS}
386- "${CMAKE_CURRENT_BINARY_DIR} /$<IF:$<BOOL:RHEL_BUILD>,libtorchvision.so,libtorchvision.so.1>" )
386+ "${CMAKE_CURRENT_BINARY_DIR} /$<IF:$<BOOL:${ RHEL_BUILD} >,libtorchvision.so,libtorchvision.so.1>" )
387387 endif () # TRITON_PYTORCH_ENABLE_TORCHVISION
388388
389389 if (${TRITON_PYTORCH_ENABLE_TORCHTRT} )
0 commit comments