Skip to content

Commit d661eba

Browse files
metascroyfacebook-github-bot
authored andcommitted
Delete copy of quantized SDPA in torchao/experimental (pytorch#13924)
Summary: X-link: pytorch/ao#2952 This deletes the SDPA/matmul kernels in torchao/experimental. These kernels were already migrated to torchao/csrc/cpu/torch_free_kernels in the first diff, but we needed to keep them around in torchao/experimental to do a pin bump in ExecuTorch OSS after the first diff lands. Test Plan: Imported from GitHub, without a `Test Plan:` line. Rollback Plan: Reviewed By: kimishpatel Differential Revision: D81640227 Pulled By: metascroy
1 parent 79744f0 commit d661eba

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

extension/llm/custom_ops/op_sdpa_impl.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
#endif
2525
#include <executorch/extension/kernel_util/make_boxed_from_unboxed_functor.h>
2626

27-
#include <torchao/experimental/kernels/cpu/interface/quantized_matmul.h>
27+
#include <torchao/csrc/cpu/torch_free_kernels/interface/quantized_matmul.h>
2828

2929
namespace torch {
3030
namespace executor {

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)