14 lines
928 B
Diff
14 lines
928 B
Diff
diff --git a/torchao/csrc/cpu/shared_kernels/Utils.cmake b/torchao/csrc/cpu/shared_kernels/Utils.cmake
|
|
index be7004784..0e1a2ed0e 100644
|
|
--- a/torchao/csrc/cpu/shared_kernels/Utils.cmake
|
|
+++ b/torchao/csrc/cpu/shared_kernels/Utils.cmake
|
|
@@ -21,7 +21,7 @@ function(target_link_torchao_parallel_backend target_name torchao_parallel_backe
|
|
target_link_libraries(${target_name} PRIVATE "${TORCH_LIBRARIES}")
|
|
|
|
target_compile_definitions(${target_name} PRIVATE TORCHAO_PARALLEL_ATEN=1 AT_PARALLEL_OPENMP=1 INTRA_OP_PARALLEL=1)
|
|
- target_link_libraries(${target_name} PRIVATE ${TORCH_INSTALL_PREFIX}/lib/libomp${CMAKE_SHARED_LIBRARY_SUFFIX})
|
|
+ target_link_libraries(${target_name} PRIVATE @openmp@/lib/libomp${CMAKE_SHARED_LIBRARY_SUFFIX})
|
|
|
|
elseif(TORCHAO_PARALLEL_BACKEND_TOUPPER STREQUAL "EXECUTORCH")
|
|
message(STATUS "Building with TORCHAO_PARALLEL_BACKEND=TORCHAO_PARALLEL_EXECUTORCH")
|