Include dependency graph for memory_allocator.cpp:
Go to the source code of this file.
Functions | |
| int * | mlir_aie_mem_alloc (aie_libxaie_ctx_t *_xaie, ext_mem_model_t &handle, int size) |
| Allocate a buffer in device memory. | |
| void | mlir_aie_sync_mem_cpu (ext_mem_model_t &handle) |
| Synchronize the buffer from the device to the host CPU. | |
| void | mlir_aie_sync_mem_dev (ext_mem_model_t &handle) |
| Synchronize the buffer from the host CPU to the device. | |
| u64 | mlir_aie_get_device_address (aie_libxaie_ctx_t *_xaie, void *VA) |
| Return a device address corresponding to the given host address. | |
Function Documentation
◆ mlir_aie_get_device_address()
| u64 mlir_aie_get_device_address | ( | aie_libxaie_ctx_t * | _xaie, |
| void * | host_address | ||
| ) |
Return a device address corresponding to the given host address.
- Parameters
-
host_address A host-side pointer returned from mlir_aie_mem_alloc
Definition at line 51 of file memory_allocator.cpp.
References aie_libxaie_ctx_t::allocations.
◆ mlir_aie_mem_alloc()
| int * mlir_aie_mem_alloc | ( | aie_libxaie_ctx_t * | _xaie, |
| ext_mem_model_t & | handle, | ||
| int | size | ||
| ) |
Allocate a buffer in device memory.
- Parameters
-
bufIdx The index of the buffer to allocate. size The number of 32-bit words to allocate
- Returns
- A host-side pointer that can write into the given buffer.
- Todo:
- This is at best a quick hack and should be replaced
Definition at line 17 of file memory_allocator.cpp.
References aie_libxaie_ctx_t::allocations, ext_mem_model_t::physicalAddr, ext_mem_model_t::size, and ext_mem_model_t::virtualAddr.
◆ mlir_aie_sync_mem_cpu()
| void mlir_aie_sync_mem_cpu | ( | ext_mem_model_t & | handle | ) |
Synchronize the buffer from the device to the host CPU.
This is expected to be called after the device writes data into device memory, so that the data can be read by the CPU. In a non-cache coherent system, this implies invalidating the processor cache associated with the buffer.
- Parameters
-
bufIdx The buffer index.
Definition at line 43 of file memory_allocator.cpp.
References ext_mem_model_t::physicalAddr, ext_mem_model_t::size, and ext_mem_model_t::virtualAddr.
◆ mlir_aie_sync_mem_dev()
| void mlir_aie_sync_mem_dev | ( | ext_mem_model_t & | handle | ) |
Synchronize the buffer from the host CPU to the device.
This is expected to be called after the host writes data into device memory, so that the data can be read by the device. In a non-cache coherent system, this implies flushing the processor cache associated with the buffer.
- Parameters
-
bufIdx The buffer index.
Definition at line 47 of file memory_allocator.cpp.
References ext_mem_model_t::physicalAddr, ext_mem_model_t::size, and ext_mem_model_t::virtualAddr.
