Skip to content

Commit 3ed3203

Browse files
metascroyfacebook-github-bot
authored andcommitted
Move CPU kernels out of experimental
Summary: This PR: * Moves torchao/experimental/kernels -> torchao/csrc/cpu/torch_free_kernels. This directory houses CPU kernels that do not depend on torch and are written with raw pointers. * Moves torchao/experimental/ops -> torchao/csrc/cpu/shared_kernels. This directory houses CPU kernels that are shared between ATen and ExecuTorch, and can be built for either runtime based on compile-time flags. In addition, it: * Fixes include paths and CMake files to make the above changes buildable. * Adjusts setup.py The PR is on the larger side, but most changes are not meaningful. For review, I suggest focusing on the changes to setup.py. X-link: pytorch/ao#2868 Reviewed By: kimishpatel Differential Revision: D80958790 Pulled By: metascroy
1 parent 4fc602b commit 3ed3203

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

extension/llm/custom_ops/targets.bzl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ def _get_quantized_sdpa_deps():
1414
if runtime.is_oss:
1515
return []
1616
else:
17-
return ["//pytorch/ao/torchao/experimental/kernels/cpu/interface:interface"]
17+
return ["//pytorch/ao/torchao/csrc/cpu/torch_free_kernels/interface:interface"]
1818

1919
def _get_quantized_preproc_flags():
2020
if runtime.is_oss:

0 commit comments

Comments
 (0)