MCPcopy Create free account
hub / github.com/FastFlowLM/FastFlowLM / setup_device

Method setup_device

src/include/npu_utils/npu_instr_utils.hpp:230–248  ·  view source on GitHub ↗

@brief setup the npu device @param device the npu device @note The function will setup the npu device @warning The function is only used for the npu sequence that is not pre-generated

Source from the content-addressed store, hash-verified

228 ///@note The function will setup the npu device
229 ///@warning The function is only used for the npu sequence that is not pre-generated
230 void setup_device(npu_device device){
231 if (device == device_npu1){
232 // Might be wrong
233 this->npu_major = 0;
234 this->npu_minor = 1;
235 this->npu_dev_gen = 1;
236 this->npu_rows = 6;
237 this->npu_cols = 4;
238 this->npu_mem_tile_rows = 1;
239 }
240 else if (device == device_npu2){
241 this->npu_major = 0;
242 this->npu_minor = 1;
243 this->npu_dev_gen = 4;
244 this->npu_rows = 6;
245 this->npu_cols = 8;
246 this->npu_mem_tile_rows = 1;
247 }
248 }
249
250
251 ///@brief parse the npu sequence

Callers 1

npu_sequenceMethod · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected