Add support to read neureka ECC-related internal registers

Author: LuigiGhionda

inc/pulp_nnx_neureka.h

@@ -59,3 +59,9 @@ int neureka_nnx_resolve_check(neureka_dev_t *dev, neureka_task_t *task);
  * Block until you can resolve the task.
  */
 void neureka_nnx_resolve_wait(neureka_dev_t *dev, neureka_task_t *task);
+
+/** neureka_nnx_read_ecc_regs
+ *
+ * Read ECC-related internal registers.
+ */
+void neureka_nnx_read_ecc_regs(neureka_dev_t *dev, uint32_t *ecc_regs);

neureka/hal/neureka_task_defs.h

@@ -44,6 +44,8 @@
 #define NEUREKA_WEIGHT_BANDWIDTH_BYTES_1x1 (NNX_NEUREKA_BANDWIDTH_1x1/8)
 #define NEUREKA_WEIGHT_BANDWIDTH_BYTES_3x3 (NNX_NEUREKA_BANDWIDTH_3x3/8)
 
+#define NEUREKA_ECC_REGS_NUM (4)
+
 /* TASK REGISTERS */
 
 // job configuration

src/pulp_nnx_neureka.c

@@ -74,3 +74,7 @@ void neureka_nnx_resolve_wait(neureka_dev_t *dev, neureka_task_t *task) {
     neureka_bsp_event_wait_and_clear();
   }
 }
+
+void neureka_nnx_read_ecc_regs(neureka_dev_t *dev, uint32_t *ecc_regs) {
+  hwpe_ecc_regs_check(&dev->hwpe_dev, ecc_regs, NEUREKA_ECC_REGS_NUM);
+}

util/hwpe.c

@@ -29,6 +29,8 @@
 #define HWPE_SOFT_CLEAR 5
 #define HWPE_SWSYNC 6
 #define HWPE_TASK_REG_OFFSET 8
+#define HWPE_PARAMS_REG_OFFSET 48
+#define HWPE_ECC_REG_OFFSET (HWPE_TASK_REG_OFFSET+HWPE_PARAMS_REG_OFFSET)
 
 inline void hwpe_reg_write(hwpe_dev_t *dev, int reg, uint32_t value) {
   dev->base_addr[reg] = value;
@@ -83,3 +85,9 @@ void hwpe_task_queue_release(hwpe_dev_t *dev) {
 uint8_t hwpe_last_task_id(hwpe_dev_t *dev) {
   return (uint8_t)hwpe_reg_read(dev, HWPE_RUNNING_JOB);
 }
+
+void hwpe_ecc_regs_check(hwpe_dev_t *dev, uint32_t *data, int len) {
+  for (int i = 0; i < len; i++) {
+      data[i] = hwpe_reg_read(dev, HWPE_ECC_REG_OFFSET + i);
+  }
+}

util/hwpe.h

@@ -39,5 +39,6 @@ void hwpe_task_queue_write_task(hwpe_dev_t *dev, uint32_t *data, int len);
 void hwpe_task_queue_release_and_run(hwpe_dev_t *dev);
 void hwpe_task_queue_release(hwpe_dev_t *dev);
 uint8_t hwpe_last_task_id(hwpe_dev_t *dev);
+void hwpe_ecc_regs_check(hwpe_dev_t *dev, uint32_t *data, int len);
 
 #endif // !__HWPE_H__