Snitch Runtime
Loading...
Searching...
No Matches
alloc_v2.h
Go to the documentation of this file.
1// Copyright 2023 ETH Zurich and University of Bologna.
2// Licensed under the Apache License, Version 2.0, see LICENSE for details.
3// SPDX-License-Identifier: Apache-2.0
4
15extern __thread snrt_allocator_t l1_allocator_v2;
16
22inline snrt_allocator_t *snrt_l1_allocator_v2() { return &l1_allocator_v2; }
23
29inline void *snrt_l1_next_v2() { return (void *)snrt_l1_allocator_v2()->next; }
30
36inline void snrt_l1_update_next_v2(void *next) {
37 snrt_l1_allocator_v2()->next = (uint32_t)next;
38}
39
45 if (snrt_l1_allocator_v2()->next > snrt_l1_allocator_v2()->end)
46 asm volatile("ecall \n");
47}
48
60inline void *snrt_l1_alloc_cluster_local(size_t size, const size_t alignment) {
61 snrt_l1_allocator_v2()->next =
62 ALIGN_UP(snrt_l1_allocator_v2()->next, alignment);
63 void *retval = snrt_l1_next_v2();
64 snrt_l1_allocator_v2()->next += size;
66 return retval;
67}
68
82inline void *snrt_l1_alloc_compute_core_local(size_t size,
83 const size_t alignment) {
84 snrt_l1_allocator_v2()->next =
85 ALIGN_UP(snrt_l1_allocator_v2()->next, alignment);
86 void *retval = snrt_l1_next_v2() + size * snrt_cluster_core_idx();
87 snrt_l1_allocator_v2()->next += size * snrt_cluster_compute_core_num();
89 return retval;
90}
91
105inline void *snrt_compute_core_local_ptr(void *ptr, uint32_t core_idx,
106 size_t size) {
107 size_t offset = (core_idx - snrt_cluster_core_idx()) * size;
108 return (void *)((uintptr_t)ptr + offset);
109}
110
123inline void *snrt_remote_l1_ptr(void *ptr, uint32_t src_cluster_idx,
124 uint32_t dst_cluster_idx) {
125 return (void *)((uintptr_t)ptr +
126 (dst_cluster_idx - src_cluster_idx) * SNRT_CLUSTER_OFFSET);
127}
128
138inline void snrt_alloc_init_v2() {
139 // Calculate end address of the heap. The top of the TCDM address space is
140 // reserved for the cluster-local storage (CLS) and the stack of every
141 // core. We further provision a safety margin of 128B. The rest of the
142 // TCDM is reserved for the heap.
143 uint32_t heap_end_addr = snrt_cls_base_addr();
144 heap_end_addr -= (1 << SNRT_LOG2_STACK_SIZE) * snrt_cluster_core_num();
145 heap_end_addr -= 128;
146 // Initialize L1 allocator
147 snrt_l1_allocator_v2()->base =
148 ALIGN_UP(snrt_l1_start_addr(), MIN_CHUNK_SIZE);
149 snrt_l1_allocator_v2()->end = heap_end_addr;
151}
void * snrt_l1_alloc_cluster_local(size_t size, const size_t alignment)
Allocate space for a variable in the cluster's L1 memory.
Definition alloc_v2.h:60
void snrt_l1_update_next_v2(void *next)
Override the L1 allocator next pointer.
Definition alloc_v2.h:36
void snrt_alloc_init_v2()
Initialize the L1 allocator.
Definition alloc_v2.h:138
void * snrt_l1_next_v2()
Get the next pointer of the L1 allocator.
Definition alloc_v2.h:29
void * snrt_remote_l1_ptr(void *ptr, uint32_t src_cluster_idx, uint32_t dst_cluster_idx)
Get a pointer to the same offset in another cluster's L1 memory.
Definition alloc_v2.h:123
void * snrt_compute_core_local_ptr(void *ptr, uint32_t core_idx, size_t size)
Get a pointer to the same variable allocated by another core.
Definition alloc_v2.h:105
void * snrt_l1_alloc_compute_core_local(size_t size, const size_t alignment)
Allocate space for N variables in the cluster's L1 memory.
Definition alloc_v2.h:82
snrt_allocator_t * snrt_l1_allocator_v2()
Get a pointer to the L1 allocator.
Definition alloc_v2.h:22
void snrt_l1_alloc_check_bounds()
Check if the allocation exceeds the allocator bounds and raise an exception if it does.
Definition alloc_v2.h:44
Definition alloc_decls.h:10