cmake: Update PT_LIBS libtrochvision references.

mc-nv · mc-nv · commit 055f7c7cb6c1 · 2025-12-02T16:14:40.000-08:00
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -199,7 +199,7 @@ endif() # TRITON_PYTORCH_NVSHMEM
 if (${TRITON_PYTORCH_ENABLE_TORCHVISION})
   set(PT_LIBS
       ${PT_LIBS}
-      $<IF:$<BOOL:${RHEL_BUILD}>,libtorchvision.so,libtorchvision.so.1>
+      libtorchvision.so.1
   )
 endif() # TRITON_PYTORCH_ENABLE_TORCHVISION
 

Original file line number	Diff line number	Diff line change
`@@ -199,7 +199,7 @@ endif() # TRITON_PYTORCH_NVSHMEM`
`199`	`199`	`if (${TRITON_PYTORCH_ENABLE_TORCHVISION})`
`200`	`200`	`set(PT_LIBS`
`201`	`201`	`${PT_LIBS}`
`202`		`- $<IF:$<BOOL:${RHEL_BUILD}>,libtorchvision.so,libtorchvision.so.1>`
	`202`	`+ libtorchvision.so.1`
`203`	`203`	`)`
`204`	`204`	`endif() # TRITON_PYTORCH_ENABLE_TORCHVISION`
`205`	`205`