void load()

in recipes/streaming_convnets/inference/inference/module/nn/backend/fbgemm/PackedGemmMatrixFP16.h [49:87]

• 33 lines of code
• 2 McCabe index (conditional complexity)

void load(
    Archive& ar,
    std::shared_ptr<fbgemm::PackedGemmMatrixFP16>& packedMatrix) {
  int numRows = 0;
  int numCols = 0;

  // The following params are unused but kept for backward compatibility
  int blockRowSize = 0;
  int lastBrow = 0;
  int blockColSize = 0;
  int numBrow = 0;
  int numBcol = 0;
  int matSize = 0;

  ar(numRows,
     numCols,
     blockRowSize,
     lastBrow,
     blockColSize,
     numBrow,
     numBcol,
     matSize);

  std::vector<fbgemm::float16> tempBufFp16;
  ar(tempBufFp16);

  std::vector<float> tempBufFp32(numRows * numCols);
  for (int i = 0; i < numRows * numCols; ++i) {
    tempBufFp32[i] = fbgemm::cpu_half2float(tempBufFp16[i]);
  }

  constexpr float alpha = 1.0;
  packedMatrix = std::make_shared<fbgemm::PackedGemmMatrixFP16>(
      fbgemm::matrix_op_t::NoTranspose,
      numRows,
      numCols,
      alpha,
      tempBufFp32.data());
}