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); }