Skip to content

Commit 2120f65

Browse files
jorgep31415pytorchmergebot
authored andcommitted
[AT-VK][EZ] Move ops to dedicated folder (pytorch#120364)
These ops are at the level of the OperatorRegistry from the previous change. All ExecuTorch ops will go here. ``` ATen/native/vulkan/graph/ops ``` They are not to be confused with the general ATen ops from `native_functions.yaml` that will continue to exist. All PyTorch ops are here. ``` ATen/native/vulkan/ops ``` To help think around this split, note that we can actually implement the latter ATen ops with the former OperatorRegistry ops, since it's currently a subset. Differential Revision: [D54030933](https://our.internmc.facebook.com/intern/diff/D54030933/) Pull Request resolved: pytorch#120364 Approved by: https://github.com/SS-JIA ghstack dependencies: pytorch#120362, pytorch#120363
1 parent 6d920dd commit 2120f65

File tree

8 files changed

+9
-7
lines changed

8 files changed

+9
-7
lines changed

aten/src/ATen/native/vulkan/graph/Functions.cpp

+3-2
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
#include <ATen/native/vulkan/impl/Arithmetic.h>
22
#include <ATen/native/vulkan/impl/Common.h>
33

4-
#include <ATen/native/vulkan/graph/Arithmetic.h>
54
#include <ATen/native/vulkan/graph/Functions.h>
6-
#include <ATen/native/vulkan/graph/Staging.h>
5+
6+
#include <ATen/native/vulkan/graph/ops/Arithmetic.h>
7+
#include <ATen/native/vulkan/graph/ops/Staging.h>
78

89
namespace at {
910
namespace native {

aten/src/ATen/native/vulkan/graph/Graph.cpp

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
#include <ATen/native/vulkan/graph/Graph.h>
2-
#include <ATen/native/vulkan/graph/Staging.h>
2+
3+
#include <ATen/native/vulkan/graph/ops/Staging.h>
34

45
namespace at {
56
namespace native {

aten/src/ATen/native/vulkan/graph/Arithmetic.cpp aten/src/ATen/native/vulkan/graph/ops/Arithmetic.cpp

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
#include <ATen/native/vulkan/impl/Common.h>
22

3-
#include <ATen/native/vulkan/graph/Arithmetic.h>
4-
#include <ATen/native/vulkan/graph/Staging.h>
3+
#include <ATen/native/vulkan/graph/ops/Arithmetic.h>
4+
#include <ATen/native/vulkan/graph/ops/Staging.h>
55

66
namespace at {
77
namespace native {

aten/src/ATen/native/vulkan/graph/Copy.cpp aten/src/ATen/native/vulkan/graph/ops/Copy.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
#include <ATen/native/vulkan/graph/Copy.h>
1+
#include <ATen/native/vulkan/graph/ops/Copy.h>
22

33
namespace at {
44
namespace native {

aten/src/ATen/native/vulkan/graph/Staging.cpp aten/src/ATen/native/vulkan/graph/ops/Staging.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
#include <ATen/native/vulkan/impl/Packing.h>
22

3-
#include <ATen/native/vulkan/graph/Staging.h>
3+
#include <ATen/native/vulkan/graph/ops/Staging.h>
44

55
namespace at {
66
namespace native {

0 commit comments

Comments
 (0)