From 1b74bcf98a88c54498423eeeb62315d3ebcbf19e Mon Sep 17 00:00:00 2001 From: Jack-Khuu Date: Tue, 4 Mar 2025 11:15:34 -0800 Subject: [PATCH] Remove libOMP linking for experimental kernels --- .github/workflows/torchao_experimental_test.yml | 2 +- torchao/experimental/Utils.cmake | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/.github/workflows/torchao_experimental_test.yml b/.github/workflows/torchao_experimental_test.yml index c38a2c5e78..76520c3e52 100644 --- a/.github/workflows/torchao_experimental_test.yml +++ b/.github/workflows/torchao_experimental_test.yml @@ -33,7 +33,7 @@ jobs: - name: Install requirements run: | conda activate venv - pip install --extra-index-url "https://download.pytorch.org/whl/nightly/cpu" torch=="2.6.0.dev20250104" + pip install --extra-index-url "https://download.pytorch.org/whl/nightly/cpu" torch=="2.7.0.dev20250131" pip install numpy pip install pytest USE_CPP=1 pip install . diff --git a/torchao/experimental/Utils.cmake b/torchao/experimental/Utils.cmake index 984c90006b..57a35e61fa 100644 --- a/torchao/experimental/Utils.cmake +++ b/torchao/experimental/Utils.cmake @@ -21,7 +21,6 @@ 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}) elseif(TORCHAO_PARALLEL_BACKEND_TOUPPER STREQUAL "EXECUTORCH") message(STATUS "Building with TORCHAO_PARALLEL_BACKEND=TORCHAO_PARALLEL_EXECUTORCH")