Skip to content

Commit a2c5f7e

Browse files
metascroyfacebook-github-bot
authored andcommitted
Delete copy of quantized SDPA in torchao/experimental
Summary: 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. Differential Revision: D81640227
1 parent b759ae8 commit a2c5f7e

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)