Skip to content

Commit 61aa50e

Browse files
Bump torchao to 02105d4 (v0.17.0-rc1) (#18595)
Update torchao to 02105d4 (v0.17.0-rc1) and add Torch cmake prefix path for TorchAO build The torchao submodule update introduces a find_package(Torch) call that requires Torch's cmake config to be discoverable. Reuse the existing add_torch_to_cmake_prefix_path() helper from tools/cmake/Utils.cmake to fix the build. Co-authored-by: Manuel Candales <42380156+manuelcandales@users.noreply.github.com>
1 parent ee7392f commit 61aa50e

2 files changed

Lines changed: 4 additions & 1 deletion

File tree

CMakeLists.txt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -846,6 +846,9 @@ if(EXECUTORCH_BUILD_KERNELS_TORCHAO)
846846
set(TORCHAO_ENABLE_ARM_NEON_DOT ON)
847847
set(TORCHAO_BUILD_KLEIDIAI ON)
848848

849+
# TorchAO requires find_package(Torch) for include dirs; add its cmake path.
850+
add_torch_to_cmake_prefix_path()
851+
849852
# TorchAO kernels look for EXECUTORCH_INCLUDE_DIRS
850853
if(DEFINED EXECUTORCH_INCLUDE_DIRS)
851854
message(FATAL_ERROR "EXECUTORCH_INCLUDE_DIRS is already defined")

third-party/ao

Submodule ao updated 202 files

0 commit comments

Comments
 (0)