Skip to content

Commit 2510204

Browse files
authored
disable i32 index for webgpu for now [no ci]
1 parent d8abd2d commit 2510204

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ggml/src/ggml-webgpu/ggml-webgpu.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1310,7 +1310,7 @@ static bool ggml_backend_webgpu_device_supports_op(ggml_backend_dev_t dev, const
13101310
break;
13111311
case GGML_OP_CPY:
13121312
case GGML_OP_SET_ROWS:
1313-
supports_op = (op->type == GGML_TYPE_F16 && op->src[0]->type == GGML_TYPE_F32);
1313+
supports_op = (op->type == GGML_TYPE_F16 && op->src[0]->type == GGML_TYPE_F32 && op->src[1]->type == GGML_TYPE_I64);
13141314
break;
13151315
case GGML_OP_GET_ROWS:
13161316
if (op->src[0]->type == GGML_TYPE_F32 || op->src[0]->type == GGML_TYPE_F16 ||

0 commit comments

Comments
 (0)