Skip to content

Commit c71e35e

Browse files
committed
fix(ggml-metal): better granularity for support bool for CUMSUM and TRI
Branch: Mamba2SSD Signed-off-by: Gabe Goodhart <ghart@us.ibm.com>
1 parent 0468b99 commit c71e35e

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

ggml/src/ggml-metal/ggml-metal-device.m

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -656,10 +656,9 @@ bool ggml_metal_device_supports_op(ggml_metal_device_t dev, const struct ggml_te
656656
case GGML_OP_COS:
657657
case GGML_OP_LOG:
658658
return ggml_is_contiguous(op->src[0]) && op->src[0]->type == GGML_TYPE_F32;
659-
case GGML_OP_CUMSUM:
660659
case GGML_OP_TRI:
661-
//DEBUG -- Refine this!
662-
return true;
660+
return ggml_is_contiguous_rows(op->src[0]);
661+
case GGML_OP_CUMSUM:
663662
case GGML_OP_SUM:
664663
case GGML_OP_SUM_ROWS:
665664
case GGML_OP_MEAN:

0 commit comments

Comments
 (0)