1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
/*
* Copyright 2025 LunarG, Inc.
* Copyright 2025 Google LLC
* SPDX-License-Identifier: MIT
*/
#include "kk_bo.h"
#include "kk_device.h"
#include "kosmickrisp/bridge/mtl_bridge.h"
#include "util/u_memory.h"
VkResult
kk_alloc_bo(struct kk_device *dev, struct vk_object_base *log_obj,
uint64_t size_B, uint64_t align_B, struct kk_bo **bo_out)
{
VkResult result = VK_SUCCESS;
// TODO_KOSMICKRISP: Probably requires handling the buffer maximum 256MB
uint64_t minimum_alignment = 0u;
mtl_heap_buffer_size_and_align_with_length(dev->mtl_handle, &size_B,
&minimum_alignment);
minimum_alignment = MAX2(minimum_alignment, align_B);
size_B = align64(size_B, minimum_alignment);
mtl_heap *handle =
mtl_new_heap(dev->mtl_handle, size_B, KK_MTL_RESOURCE_OPTIONS);
if (handle == NULL) {
result = vk_errorf(log_obj, VK_ERROR_OUT_OF_DEVICE_MEMORY, "%m");
goto fail_heap;
}
mtl_buffer *map = mtl_new_buffer_with_length(handle, size_B, 0u);
if (map == NULL) {
result = vk_errorf(log_obj, VK_ERROR_OUT_OF_DEVICE_MEMORY, "%m");
goto fail_map;
}
struct kk_bo *bo = CALLOC_STRUCT(kk_bo);
if (bo == NULL) {
result = vk_errorf(log_obj, VK_ERROR_OUT_OF_HOST_MEMORY, "%m");
goto fail_alloc;
}
bo->mtl_handle = handle;
bo->size_B = size_B;
bo->map = map;
bo->gpu = mtl_buffer_get_gpu_address(map);
bo->cpu = mtl_get_contents(map);
*bo_out = bo;
return result;
fail_alloc:
mtl_release(map);
fail_map:
mtl_release(handle);
fail_heap:
return result;
}
void
kk_destroy_bo(struct kk_device *dev, struct kk_bo *bo)
{
mtl_release(bo->map);
mtl_release(bo->mtl_handle);
FREE(bo);
}
|
