WEAK void halide_dev_malloc(buffer_t* buf) {
if (buf->dev) {
halide_assert(halide_validate_dev_pointer(buf));
return;
}
size_t size = __buf_size(buf);
#ifdef DEBUG
halide_printf("dev_malloc allocating buffer of %zd bytes, extents: %zdx%zdx%zdx%zd strides: %zdx%zdx%zdx%zd (%d bytes per element)\n",
size, buf->extent[0], buf->extent[1], buf->extent[2], buf->extent[3],
buf->stride[0], buf->stride[1], buf->stride[2], buf->stride[3],
buf->elem_size);
#endif
buf->dev = (uint64_t)__dev_malloc(size);
halide_assert(buf->dev);
}
WEAK void halide_dev_malloc(void *user_context, buffer_t* buf) {
if (buf->dev) {
halide_assert(user_context, halide_validate_dev_pointer(user_context, buf));
return;
}
size_t size = __buf_size(user_context, buf);
#ifdef DEBUG
halide_printf(user_context, "dev_malloc allocating buffer of %lld bytes, "
"extents: %lldx%lldx%lldx%lld strides: %lldx%lldx%lldx%lld (%d bytes per element)\n",
(long long)size, (long long)buf->extent[0], (long long)buf->extent[1],
(long long)buf->extent[2], (long long)buf->extent[3],
(long long)buf->stride[0], (long long)buf->stride[1],
(long long)buf->stride[2], (long long)buf->stride[3],
buf->elem_size);
#endif
buf->dev = (uint64_t)__dev_malloc(user_context, size);
#ifdef DEBUG
halide_printf(user_context, "dev_malloc allocated buffer %p of with buf->dev of %p\n",
buf, (void *)buf->dev);
#endif
halide_assert(user_context, buf->dev);
}
