Skip to content

Commit e40ada3

Browse files
authored
[Metal] Fix depthwise conv 1D kernel name for large variant (#3289)
1 parent 38ad257 commit e40ada3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mlx/backend/metal/conv.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1041,8 +1041,8 @@ void depthwise_conv_1D_gpu(
10411041
concatenate(
10421042
base_name,
10431043
"depthwise_conv_1d_",
1044-
large ? "_large" : "",
1045-
type_to_name(out));
1044+
type_to_name(out),
1045+
large ? "_large" : "");
10461046

10471047
auto& compute_encoder = d.get_command_encoder(s.index);
10481048
auto kernel = d.get_kernel(base_name);

0 commit comments

Comments
 (0)