diff --git a/backends/vulkan/runtime/graph/ops/glsl/slice_channel.glsl b/backends/vulkan/runtime/graph/ops/glsl/slice_packed_dim.glsl similarity index 100% rename from backends/vulkan/runtime/graph/ops/glsl/slice_channel.glsl rename to backends/vulkan/runtime/graph/ops/glsl/slice_packed_dim.glsl diff --git a/backends/vulkan/runtime/graph/ops/glsl/slice_channel.yaml b/backends/vulkan/runtime/graph/ops/glsl/slice_packed_dim.yaml similarity index 80% rename from backends/vulkan/runtime/graph/ops/glsl/slice_channel.yaml rename to backends/vulkan/runtime/graph/ops/glsl/slice_packed_dim.yaml index 56317260170..718e7316824 100644 --- a/backends/vulkan/runtime/graph/ops/glsl/slice_channel.yaml +++ b/backends/vulkan/runtime/graph/ops/glsl/slice_packed_dim.yaml @@ -1,4 +1,4 @@ -slice_channel: +slice_packed_dim: parameter_names_with_default_values: DTYPE: float NDIM: 3 @@ -8,4 +8,4 @@ slice_channel: - VALUE: half - VALUE: float shader_variants: - - NAME: slice_channel + - NAME: slice_packed_dim diff --git a/backends/vulkan/runtime/graph/ops/glsl/slice_batch_height_width.glsl b/backends/vulkan/runtime/graph/ops/glsl/slice_unpacked_dim.glsl similarity index 100% rename from backends/vulkan/runtime/graph/ops/glsl/slice_batch_height_width.glsl rename to backends/vulkan/runtime/graph/ops/glsl/slice_unpacked_dim.glsl diff --git a/backends/vulkan/runtime/graph/ops/glsl/slice_batch_height_width.yaml b/backends/vulkan/runtime/graph/ops/glsl/slice_unpacked_dim.yaml similarity index 72% rename from backends/vulkan/runtime/graph/ops/glsl/slice_batch_height_width.yaml rename to backends/vulkan/runtime/graph/ops/glsl/slice_unpacked_dim.yaml index 9e69b09a304..0453bb707b1 100644 --- a/backends/vulkan/runtime/graph/ops/glsl/slice_batch_height_width.yaml +++ b/backends/vulkan/runtime/graph/ops/glsl/slice_unpacked_dim.yaml @@ -1,4 +1,4 @@ -slice_batch_height_width: +slice_unpacked_dim: parameter_names_with_default_values: DTYPE: float NDIM: 3 @@ -7,4 +7,4 @@ slice_batch_height_width: - VALUE: half - VALUE: float shader_variants: - - NAME: slice_batch_height_width + - NAME: slice_unpacked_dim diff --git a/backends/vulkan/runtime/graph/ops/impl/Slice.cpp b/backends/vulkan/runtime/graph/ops/impl/Slice.cpp index b0b2a740d6b..efda6e04992 100644 --- a/backends/vulkan/runtime/graph/ops/impl/Slice.cpp +++ b/backends/vulkan/runtime/graph/ops/impl/Slice.cpp @@ -83,7 +83,7 @@ void add_slice_tensor_copy_node( // if slice dim is the same as the packed dim, we can use the channel slice if (dim_index == packed_dim_idx) { // slice by channel - std::string kernel_name = "slice_channel"; + std::string kernel_name = "slice_packed_dim"; kernel_name.reserve(kShaderNameReserve); add_dtype_suffix(kernel_name, *t_out); @@ -119,7 +119,7 @@ void add_slice_tensor_copy_node( in_channel_stride = utils::div_up_4(in_channel_stride); } - std::string kernel_name = "slice_batch_height_width"; + std::string kernel_name = "slice_unpacked_dim"; kernel_name.reserve(kShaderNameReserve); add_dtype_suffix(kernel_name, *t_out);