Skip to content

Commit bb9949b

Browse files
committed
apply review: change kernel name of pool_2d
Signed-off-by: Junhee Yoo <[email protected]>
1 parent bd86c4c commit bb9949b

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

ggml/src/ggml-metal.m

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -722,8 +722,8 @@ @implementation GGMLMetalClass
722722
GGML_METAL_ADD_KERNEL(GGML_METAL_KERNEL_TYPE_SIN, sin, true);
723723
GGML_METAL_ADD_KERNEL(GGML_METAL_KERNEL_TYPE_COS, cos, true);
724724
GGML_METAL_ADD_KERNEL(GGML_METAL_KERNEL_TYPE_SUM_ROWS, sum_rows, true);
725-
GGML_METAL_ADD_KERNEL(GGML_METAL_KERNEL_TYPE_POOL_2D_AVG_F32, avg_pool_2d_f32, true);
726-
GGML_METAL_ADD_KERNEL(GGML_METAL_KERNEL_TYPE_POOL_2D_MAX_F32, max_pool_2d_f32, true);
725+
GGML_METAL_ADD_KERNEL(GGML_METAL_KERNEL_TYPE_POOL_2D_AVG_F32, pool_2d_avg_f32, true);
726+
GGML_METAL_ADD_KERNEL(GGML_METAL_KERNEL_TYPE_POOL_2D_MAX_F32, pool_2d_max_f32, true);
727727
}
728728

729729
[metal_library release];

ggml/src/ggml-metal.metal

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6452,7 +6452,7 @@ template [[host_name("kernel_mul_mv_id_iq2_s_f32")]] kernel kernel_mul_mv_id_t
64526452
template [[host_name("kernel_mul_mv_id_iq4_nl_f32")]] kernel kernel_mul_mv_id_t kernel_mul_mv_id<mmv_fn<kernel_mul_mv_iq4_nl_f32_impl>>;
64536453
template [[host_name("kernel_mul_mv_id_iq4_xs_f32")]] kernel kernel_mul_mv_id_t kernel_mul_mv_id<mmv_fn<kernel_mul_mv_iq4_xs_f32_impl>>;
64546454

6455-
kernel void kernel_max_pool_2d_f32(
6455+
kernel void kernel_pool_2d_max_f32(
64566456
device const float * src0,
64576457
device float * dst,
64586458
constant int32_t & k0,
@@ -6498,7 +6498,7 @@ kernel void kernel_max_pool_2d_f32(
64986498
o_ptr[cur_oh * OW + cur_ow] = res;
64996499
}
65006500

6501-
kernel void kernel_avg_pool_2d_f32(
6501+
kernel void kernel_pool_2d_avg_f32(
65026502
device const float* src0,
65036503
device float* dst,
65046504
constant int32_t& k0,

0 commit comments

Comments
 (0)