Skip to content

Commit 7a1703a

Browse files
committed
Update on "[5/n][ET-VK][Ops] aten.flip"
Port from LiteInterpreter's [`flip.glsl`](https://github.com/pytorch/pytorch/blob/main/aten/src/ATen/native/vulkan/glsl/flip.glsl). ``` - func: flip(Tensor self, int[] dims) -> Tensor ``` Will use this to verify AHB image is interpreted correctly. Differential Revision: [D63843843](https://our.internmc.facebook.com/intern/diff/D63843843/) [ghstack-poisoned]
2 parents bd36872 + d1d6444 commit 7a1703a

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

backends/vulkan/runtime/graph/ops/impl/Flip.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,8 +65,8 @@ void add_flip_node(
6565
graph.create_local_wg_size(out),
6666
// Inputs and Outputs
6767
{
68-
{out, vkapi::MemoryAccessType::WRITE},
69-
{in, vkapi::MemoryAccessType::READ},
68+
{out, vkapi::kWrite},
69+
{in, vkapi::kRead},
7070
},
7171
// Parameter buffers
7272
{

backends/vulkan/test/op_tests/cases.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1163,7 +1163,7 @@ def get_squeeze_copy_dim_inputs():
11631163

11641164
@register_test_suite("aten.flip.default")
11651165
def get_flip_inputs():
1166-
Test = namedtuple("VkIndexSelectTest", ["self", "dim"])
1166+
Test = namedtuple("Flip", ["self", "dim"])
11671167
Test.__new__.__defaults__ = (None, 0)
11681168

11691169
test_cases = [

0 commit comments

Comments
 (0)