19 lines
306 B
C++
19 lines
306 B
C++
#ifndef _COMMON_H_
|
|
#define _COMMON_H_
|
|
|
|
#include <cstdint>
|
|
|
|
#define KERNEL_ARG_DEV_MEM_ADDR 0x9fff0000
|
|
#define DEV_SMEM_START_ADDR 0xff000000
|
|
|
|
typedef struct {
|
|
uint32_t dim_seqlen;
|
|
uint32_t dim_headdim;
|
|
uint64_t addr_q;
|
|
uint64_t addr_k;
|
|
uint64_t addr_v;
|
|
uint64_t addr_o;
|
|
} kernel_arg_t;
|
|
|
|
#endif
|