Skip to content

Commit 2e677bb

Browse files
committed
Removing generation expression
1 parent d3b9a0b commit 2e677bb

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

CMakeLists.txt

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -161,9 +161,15 @@ set(PT_LIBS
161161
"libtorch_cuda_linalg.so"
162162
"libtorch_global_deps.so"
163163
"libjpeg.so.62"
164-
$<IF:$<BOOL:${TRITON_PYTORCH_NVSHMEM}>,libtorch_nvshmem.so,>
165164
)
166165

166+
if (${TRITON_PYTORCH_NVSHMEM})
167+
set(PT_LIBS
168+
${PT_LIBS}
169+
"libtorch_nvshmem.so"
170+
)
171+
endif() # TRITON_PYTORCH_NVSHMEM
172+
167173
if (${TRITON_PYTORCH_ENABLE_TORCHVISION})
168174
set(PT_LIBS
169175
${PT_LIBS}

0 commit comments

Comments
 (0)