abnerhexu
8f7dba5920
- Add kernels/blackwell_insts/ with test kernel and input data - Update linker script with extended memory layout - Remove obsolete sgemm_tcore_blackwell Makefile - Update VX_types.h and common.h
19 lines
318 B
C++
19 lines
318 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 __attribute__((packed)) {
|
|
uint32_t dim_m;
|
|
uint32_t dim_n;
|
|
uint32_t dim_k;
|
|
uint64_t addr_a;
|
|
uint64_t addr_b;
|
|
uint64_t addr_c;
|
|
} kernel_arg_t;
|
|
|
|
#endif
|