Commit 79e1285
Move CPU kernels out of experimental (pytorch#13920)
Summary:
Pull Request resolved: pytorch#13920
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
Test Plan: CI tests
Reviewed By: kimishpatel
Differential Revision: D80958790
Pulled By: metascroy1 parent ab44d06 commit 79e1285
2 files changed
+2
-2
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
24 | 24 | | |
25 | 25 | | |
26 | 26 | | |
27 | | - | |
| 27 | + | |
28 | 28 | | |
29 | 29 | | |
30 | 30 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
14 | 14 | | |
15 | 15 | | |
16 | 16 | | |
17 | | - | |
| 17 | + | |
18 | 18 | | |
19 | 19 | | |
20 | 20 | | |
| |||
0 commit comments