Skip to content

Commit 14b080f

Browse files
hexagon: add asserts to set/get_tensor to make sure we handle complete tensors
Same asserts as the CPU backend.
1 parent 7f03a40 commit 14b080f

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

ggml/src/ggml-hexagon/ggml-hexagon.cpp

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1286,6 +1286,9 @@ static void ggml_backend_hexagon_buffer_set_tensor(ggml_backend_buffer_t buffer,
12861286
const void * data,
12871287
size_t offset,
12881288
size_t size) {
1289+
GGML_ASSERT(offset == 0);
1290+
GGML_ASSERT(size == ggml_nbytes(tensor));
1291+
12891292
auto ctx = (ggml_backend_hexagon_buffer_context *) buffer->context;
12901293
auto sess = ctx->sess;
12911294

@@ -1319,6 +1322,9 @@ static void ggml_backend_hexagon_buffer_get_tensor(ggml_backend_buffer_t buffer,
13191322
void * data,
13201323
size_t offset,
13211324
size_t size) {
1325+
GGML_ASSERT(offset == 0);
1326+
GGML_ASSERT(size == ggml_nbytes(tensor));
1327+
13221328
auto ctx = (ggml_backend_hexagon_buffer_context *) buffer->context;
13231329
auto sess = ctx->sess;
13241330

0 commit comments

Comments
 (0)