Skip to content

Commit 6133476

Browse files
committed
Update on "Remove ExecuTorch copy of Vectorized"
All uses are outside ExecuTorch core, so we can just use ATen Vectorized. Differential Revision: [D66396016](https://our.internmc.facebook.com/intern/diff/D66396016/) [ghstack-poisoned]
2 parents 98acc2f + 555e917 commit 6133476

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

extension/llm/custom_ops/op_sdpa_impl.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,9 +8,9 @@
88

99
#pragma once
1010

11+
#include <ATen/cpu/vec/vec.h>
1112
#include <executorch/kernels/optimized/blas/CPUBlas.h>
1213
#include <executorch/kernels/optimized/vec/functional.h>
13-
#include <executorch/kernels/optimized/vec/vec.h>
1414
#include <executorch/runtime/core/exec_aten/util/dim_order_util.h>
1515
// @lint-ignore CLANGTIDY facebook-unused-include-check
1616
#include <executorch/runtime/core/exec_aten/util/scalar_type_util.h>
@@ -319,7 +319,7 @@ void _qk_at_v_gemm(
319319
constexpr size_t kKVDim = 4;
320320

321321
template <typename T>
322-
inline void _store(T* dst, ::executorch::vec::Vectorized<T> src) {
322+
inline void _store(T* dst, ::at::vec::Vectorized<T> src) {
323323
src.store(dst);
324324
}
325325

@@ -354,7 +354,7 @@ inline double calculate_scale(const Tensor& query, optional<double> scale) {
354354
return softmax_scale;
355355
}
356356

357-
namespace vec = ::executorch::vec;
357+
namespace vec = ::at::vec;
358358
using Tensor = ::executorch::aten::Tensor;
359359

360360
// 1) out = exp(a - val)

0 commit comments

Comments
 (0)