wu-arch/kernels
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

adding texture test to tex_demo

Browse Source
This commit is contained in:
Blaise Tine
2021-03-17 09:52:33 -04:00
parent 676a13f30d
commit 2cbc1c4161
17 changed files with 654 additions and 632 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
driver/tests/tex_demo/kernel.c
View File

@@ -2,30 +2,56 @@
#include <vx_intrinsics.h>
#include <vx_tex.h>
#include "common.h"
struct tile_arg_t {
struct kernel_arg_t karg;
uint32_t tile_width;
uint32_t tile_height;
float deltaX;
float deltaY;
};
void kernel_body(int task_id, void* arg) {
struct kernel_arg_t* _arg = (struct kernel_arg_t*)(arg);
uint32_t count = _arg->task_size;
// int32_t* src0_ptr = (int32_t*)_arg->src0_ptr;
// int32_t* src1_ptr = (int32_t*)_arg->src1_ptr;
int32_t* dst_ptr = (int32_t*)_arg->dst_ptr;
struct tile_arg_t* _arg = (struct tile_arg_t*)(arg);
uint32_t xoffset = task_id * _arg->tile_width;
uint32_t yoffset = task_id * _arg->tile_height;
uint32_t* dst_ptr = (uint32_t*)_arg->karg.dst_ptr + xoffset + yoffset * _arg->karg.dst_pitch;
unsigned lod = 1;
unsigned u = 1;
unsigned v = 1;
unsigned t = 1;
float fu = xoffset * _arg->deltaX;
float fv = yoffset * _arg->deltaY;
// vx_csr_set(_arg->device_ptr, 0, 0xfd0, 0xffd);
uint32_t offset = task_id * count;
for (uint32_t i = 0; i < count; ++i) {
// dst_ptr[offset+i] = src0_ptr[offset+i] + src1_ptr[offset+i];
dst_ptr[offset+i] = vx_tex(t, u, v, lod);
for (uint32_t y = 0; y < _arg->tile_height; ++y) {
for (uint32_t x = 0; x < _arg->tile_width; ++x) {
int32_t u = (int32_t)(fu * (1<<28));
int32_t v = (int32_t)(fv * (1<<28));
dst_ptr[x] = vx_tex(0, u, v, 0);
fu += _arg->deltaX;
}
dst_ptr += _arg->karg.dst_pitch;
fv += _arg->deltaY;
}
}
int main() {
struct kernel_arg_t* arg = (struct kernel_arg_t*)KERNEL_ARG_DEV_MEM_ADDR;
vx_spawn_tasks(arg->num_tasks, kernel_body, arg);
struct kernel_arg_t* arg = (struct kernel_arg_t*)0x0;
// configure texture unit
vx_csr_write(CSR_TEX0_ADDR, arg->src_ptr);
vx_csr_write(CSR_TEX0_FORMAT, 0);
vx_csr_write(CSR_TEX0_WIDTH, arg->src_width);
vx_csr_write(CSR_TEX0_HEIGHT, arg->src_height);
vx_csr_write(CSR_TEX0_PITCH, arg->src_pitch);
vx_csr_write(CSR_TEX0_WRAP_U, 0);
vx_csr_write(CSR_TEX0_WRAP_V, 0);
vx_csr_write(CSR_TEX0_MIN_FILTER, 0);
vx_csr_write(CSR_TEX0_MAX_FILTER, 0);
struct tile_arg_t targ;
targ.karg = *arg;
targ.tile_width = arg->dst_width;
targ.tile_height = (arg->dst_height + arg->num_tasks - 1) / arg->num_tasks;
targ.deltaX = 1.0f / arg->dst_width;
targ.deltaY = 1.0f / arg->dst_height;
vx_spawn_tasks(arg->num_tasks, kernel_body, targ);
}