Skip to content

Commit

Permalink
metal : add missing asserts (llama/7617)
Browse files Browse the repository at this point in the history
  • Loading branch information
ggerganov committed Jun 15, 2024
1 parent 3429c39 commit 0999c76
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions src/ggml-kompute.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1597,8 +1597,8 @@ static void ggml_vk_graph_compute(struct ggml_kompute_context * ctx, struct ggml
{
GGML_ASSERT(ne00 == ne10);

ggml_is_contiguous_2(src0);
ggml_is_contiguous_2(src1);
GGML_ASSERT(ggml_is_contiguous_2(src0));
GGML_ASSERT(ggml_is_contiguous_2(src1));

GGML_ASSERT(ne12 % ne02 == 0);
GGML_ASSERT(ne13 % ne03 == 0);
Expand Down
4 changes: 2 additions & 2 deletions src/ggml-metal.m
Original file line number Diff line number Diff line change
Expand Up @@ -1519,8 +1519,8 @@ static enum ggml_status ggml_metal_graph_compute(
{
GGML_ASSERT(ne00 == ne10);

ggml_is_contiguous_2(src0);
ggml_is_contiguous_2(src1);
GGML_ASSERT(ggml_is_contiguous_2(src0));
GGML_ASSERT(ggml_is_contiguous_2(src1));

GGML_ASSERT(ne12 % ne02 == 0);
GGML_ASSERT(ne13 % ne03 == 0);
Expand Down

0 comments on commit 0999c76

Please sign in to comment.