File tree Expand file tree Collapse file tree 1 file changed +10
-10
lines changed
Expand file tree Collapse file tree 1 file changed +10
-10
lines changed Original file line number Diff line number Diff line change @@ -176,16 +176,16 @@ class NeutronBackend final : public PyTorchBackendInterface {
176176 // cfg->mcfg.microcode
177177 // cfg->mcfg.weights
178178 // cfg->mcfg.kernels
179- const uint8_t * transpositionFlags = static_cast <const uint8_t *>(processed->data ());
180- uint32_t numInputs = transpositionFlags [INPUT_TENSOR_FORMAT_LEN_POS];
181- uint32_t numOutputs = transpositionFlags [OUTPUT_TENSOR_FORMAT_LEN_POS];
182- cfg->numInputArgs = transpositionFlags [INPUT_ARGS_LEN_POS];
183- cfg->inputTranspositionFlags = INPUT_TENSOR_FORMAT_ARRAY_ADDR (transpositionFlags );
184- cfg->outputTranspositionFlags = OUTPUT_TENSOR_FORMAT_ARRAY_ADDR (transpositionFlags );
185- cfg->inputMap = INPUT_TENSOR_MAP_ARRAY_ADDR (transpositionFlags );
186- cfg->outputMap = OUTPUT_TENSOR_MAP_ARRAY_ADDR (transpositionFlags );
187-
188- const uint32_t * buffer = static_cast <const uint32_t *>(static_cast <const void *>PAYLOAD_ADDR (transpositionFlags ));
179+ const uint8_t * payloadFlags = static_cast <const uint8_t *>(processed->data ());
180+ uint32_t numInputs = payloadFlags [INPUT_TENSOR_FORMAT_LEN_POS];
181+ uint32_t numOutputs = payloadFlags [OUTPUT_TENSOR_FORMAT_LEN_POS];
182+ cfg->numInputArgs = payloadFlags [INPUT_ARGS_LEN_POS];
183+ cfg->inputTranspositionFlags = INPUT_TENSOR_FORMAT_ARRAY_ADDR (payloadFlags );
184+ cfg->outputTranspositionFlags = OUTPUT_TENSOR_FORMAT_ARRAY_ADDR (payloadFlags );
185+ cfg->inputMap = INPUT_TENSOR_MAP_ARRAY_ADDR (payloadFlags );
186+ cfg->outputMap = OUTPUT_TENSOR_MAP_ARRAY_ADDR (payloadFlags );
187+
188+ const uint32_t * buffer = static_cast <const uint32_t *>(static_cast <const void *>PAYLOAD_ADDR (payloadFlags ));
189189 uint32_t magicWord = buffer[0 ];
190190 // Check valid microcode.
191191 if (magicWord != 0x64434D6E ) {
You can’t perform that action at this time.
0 commit comments