Commit 3ed3203
Move CPU kernels out of experimental
Summary:
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
Reviewed By: kimishpatel
Differential Revision: D80958790
Pulled By: metascroy1 parent 4fc602b commit 3ed3203
1 file changed
+1
-1
lines changed| 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