minor update

This commit is contained in:
Blaise Tine
2021-05-06 08:55:46 -07:00
parent 6107bf8247
commit 3e88a71801
6 changed files with 560 additions and 510 deletions

View File

@@ -7,17 +7,17 @@ void kernel_body(int task_id, void* arg) {
struct kernel_arg_t* _arg = (struct kernel_arg_t*)(arg);
uint32_t stride = _arg->stride;
uint32_t* addr_ptr = (uint32_t*)_arg->addr_ptr;
int32_t* src_ptr = (int32_t*)_arg->src_ptr;
int32_t* dst_ptr = (int32_t*)_arg->dst_ptr;
float* src_ptr = (float*)_arg->src_ptr;
float* dst_ptr = (float*)_arg->dst_ptr;
uint32_t offset = task_id * stride;
for (uint32_t i = 0; i < stride; ++i) {
int value = 0;
float value = 0.0f;
for (uint32_t j = 0; j < NUM_LOADS; ++j) {
uint32_t addr = offset + i + j;
uint32_t index = addr_ptr[addr];
value += src_ptr[index];
value *= src_ptr[index];
}
dst_ptr[offset+i] = value;
}