From ba0b1f73a842d64c483a28759e3423f1f3566f7f Mon Sep 17 00:00:00 2001 From: Sunidhi-Gaonkar1 Date: Wed, 17 Sep 2025 09:24:24 +0530 Subject: [PATCH] Added power CPU support. --- CMakeLists.txt | 30 ++++++++++++++++++++---------- 1 file changed, 20 insertions(+), 10 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 3afe90b..2bf9b18 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -151,16 +151,26 @@ endif() # TRITON_ENABLE_NVTX # configure_file(src/libtriton_pytorch.ldscript libtriton_pytorch.ldscript COPYONLY) -set(PT_LIBS - "libc10.so" - "libc10_cuda.so" - "libtorch.so" - "libtorch_cpu.so" - "libtorch_cuda.so" - "libtorch_cuda_linalg.so" - "libtorch_global_deps.so" - "libjpeg.so.62" -) +if(NOT CMAKE_SYSTEM_PROCESSOR STREQUAL "ppc64le") + set(PT_LIBS + "libc10.so" + "libc10_cuda.so" + "libtorch.so" + "libtorch_cpu.so" + "libtorch_cuda.so" + "libtorch_cuda_linalg.so" + "libtorch_global_deps.so" + "libjpeg.so.62" + ) + else() + set(PT_LIBS + "libc10.so" + "libtorch.so" + "libtorch_cpu.so" + "libtorch_global_deps.so" + "libjpeg.so.62" + ) +endif() if (${TRITON_PYTORCH_ENABLE_TORCHVISION}) set(PT_LIBS