| #ifndef BSH_GEMM_ALLOCATOR_H |
| #define BSH_GEMM_ALLOCATOR_H |
|
|
| #include <stddef.h> |
| #include <stdint.h> |
| #include <stdbool.h> |
|
|
| #define CACHE_LINE_SIZE 64 |
| #define SIMD_ALIGNMENT 64 |
| #define DEFAULT_ALIGNMENT 32 |
|
|
| typedef struct { |
| void* ptr; |
| size_t size; |
| size_t alignment; |
| bool is_aligned; |
| } aligned_buffer_t; |
|
|
| aligned_buffer_t* buffer_allocate(size_t size, size_t alignment); |
| void buffer_free(aligned_buffer_t* buf); |
| void* buffer_ptr(aligned_buffer_t* buf); |
| bool buffer_is_aligned(aligned_buffer_t* buf, size_t required_alignment); |
|
|
| typedef struct { |
| float* data; |
| size_t rows; |
| size_t cols; |
| size_t stride; |
| size_t alignment; |
| bool owns_data; |
| } matrix_t; |
|
|
| matrix_t* matrix_allocate(size_t rows, size_t cols, size_t alignment); |
| void matrix_free(matrix_t* mat); |
| float* matrix_ptr(matrix_t* mat); |
| float matrix_get(const matrix_t* mat, size_t i, size_t j); |
| void matrix_set(matrix_t* mat, size_t i, size_t j, float val); |
| void matrix_zero(matrix_t* mat); |
| void matrix_fill(matrix_t* mat, float val); |
| void matrix_copy(const matrix_t* src, matrix_t* dst); |
|
|
| #endif |
|
|
