Skip to content

Commit 7f92671

Browse files
authored
Fix * 4 debug fix that got accidentally landed in op_sdpa.cpp (#9492)
As the title. Test: CI ### Summary [PLEASE REMOVE] See [CONTRIBUTING.md's Pull Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests) for ExecuTorch PR guidelines. [PLEASE REMOVE] If this PR closes an issue, please add a `Fixes #<issue-id>` line. [PLEASE REMOVE] If this PR introduces a fix or feature that should be the upcoming release notes, please add a "Release notes: <area>" label. For a list of available release notes labels, check out [CONTRIBUTING.md's Pull Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests). ### Test plan [PLEASE REMOVE] How did you test this PR? Please write down any manual commands you used and note down tests that you have written if applicable.
1 parent d980ce0 commit 7f92671

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

extension/llm/custom_ops/op_sdpa.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -382,7 +382,7 @@ void cpu_flash_attention(
382382
/* qk_sum */ qSplitSize +
383383
/* dst */ qSplitSize * headSize;
384384

385-
int64_t size_bytes = size_per_thread * num_thread * query.element_size() * 4;
385+
int64_t size_bytes = size_per_thread * num_thread * query.element_size();
386386
std::vector<char> buf_vec(size_bytes);
387387
void* buf = reinterpret_cast<void*>(buf_vec.data());
388388
// Need to double check the following

0 commit comments

Comments
 (0)