ggml-alloc: >= when checking for out-of-bounds

Co-authored-by: slaren <slarengh@gmail.com>
This commit is contained in:
Sam Spilsbury 2023-08-09 23:33:33 +03:00 committed by GitHub
parent 1b7f2c6dab
commit b810424edf
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -397,7 +397,7 @@ static void allocate_node(struct ggml_allocr * alloc, struct ggml_tensor * node)
// if the node's data is external, then we cannot re-use it
if ((char *) parent->data < (char *) alloc->data ||
(char *) parent->data > ((char *) alloc->data + alloc->size)) {
(char *) parent->data >= ((char *) alloc->data + alloc->size)) {
AT_PRINTF("not reusing parent %s for %s as %p is external\n", parent->name, node->name, parent->data);
continue;
}