Make the KWS model working on RT700.

Author: jirioc

backends/nxp/runtime/NeutronBackend.cpp

@@ -51,6 +51,7 @@ typedef struct {
     int numInputs = 0;
     int numOutputs = 0;
     int numInputArgs = 0;
+    uint32_t scratchSize = 0;
     NeutronModelConfig mcfg;
     NeutronDataConfig dcfg;
     NeutronModelHandle nmh = NULL;
@@ -194,6 +195,7 @@ class NeutronBackend final : public PyTorchBackendInterface {
     }
     uint32_t microcodeSize = buffer[6];
     uint32_t weightsSize = buffer[7];
+    cfg->scratchSize = buffer[9];
     cfg->numInputs = buffer[11];
     cfg->numOutputs = buffer[12];
     if (cfg->numInputs != numInputs) {
@@ -236,7 +238,7 @@ class NeutronBackend final : public PyTorchBackendInterface {
     // Allocate place for input and output pointers.
     cfg->dcfg.inputs = static_cast<const void**>(context.allocate(cfg->numInputs * sizeof(void*)));
     cfg->dcfg.outputs = static_cast<void**>(context.allocate(cfg->numOutputs * sizeof(void*)));
-    cfg->dcfg.outputs[cfg->numOutputs] = static_cast<void**>(context.allocate(1 * sizeof(void*)));
+    cfg->dcfg.outputs[cfg->numOutputs] = static_cast<void*>(context.allocate(cfg->scratchSize, 16));
 
     // Set inputs from args.
     // Transpose inputs if needed.