Skip to content

Commit 2412bb0

Browse files
committed
update ggml_backend_vk_device_supports_op
1 parent 6b6eede commit 2412bb0

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

ggml/src/ggml-vulkan/ggml-vulkan.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12055,6 +12055,10 @@ static bool ggml_backend_vk_device_supports_op(ggml_backend_dev_t dev, const ggm
1205512055
case GGML_OP_CONCAT:
1205612056
case GGML_OP_SCALE:
1205712057
case GGML_OP_PAD:
12058+
return (ggml_get_op_params_i32(op, 0) == 0) && (ggml_get_op_params_i32(op, 1) == 0) &&
12059+
(ggml_get_op_params_i32(op, 2) == 0) && (ggml_get_op_params_i32(op, 3) == 0) &&
12060+
(ggml_get_op_params_i32(op, 4) == 0) && (ggml_get_op_params_i32(op, 5) == 0) &&
12061+
(ggml_get_op_params_i32(op, 6) == 0) && (ggml_get_op_params_i32(op, 7) == 0);
1205812062
case GGML_OP_ROLL:
1205912063
case GGML_OP_DIAG_MASK_INF:
1206012064
case GGML_OP_SOFT_MAX:

0 commit comments

Comments
 (0)