Method MakePrepackedMatrices

tensorflow/lite/experimental/ruy/test.h:1606–1652

template <typename LhsScalar, typename RhsScalar, typename SpecType>
void TestSet<LhsScalar, RhsScalar, SpecType>::MakePrepackedMatrices() {
  RUY_CHECK(life_stage == LifeStage::kHasResultPaths);

  // Prepacked matrices are Path-dependent, so create them for each test result.
  for (auto& result : results) {
    // If this result uses an external path, then skip this entirely.
    if (result->path == Path::kNone) {
      continue;
    }
    // Pre-packing doesn't make sense for Path::kReference.
    // TODO(silvasean): Make Path::kReference an ExternalPath?
    if (result->path == Path::kReference) {
      continue;
    }

    // Determine whether we should create/use prepacked matrices.
    if (benchmark) {
      // For benchmarking, do as requested.
      result->use_prepacked_lhs = benchmark_prepack_lhs;
      result->use_prepacked_rhs = benchmark_prepack_rhs;
    } else {
      // When testing, randomly pre-pack sometimes. But don't do it too often.
      result->use_prepacked_lhs = (global_random_engine()() & 7) == 0;
      result->use_prepacked_rhs = (global_random_engine()() & 7) == 0;
    }

    // Create the pre-packed matrices.
    PrepackedMatrix* prepacked_lhs_ptr =
        result->use_prepacked_lhs ? &result->prepacked_lhs : nullptr;
    PrepackedMatrix* prepacked_rhs_ptr =
        result->use_prepacked_rhs ? &result->prepacked_rhs : nullptr;
    auto alloc_fn = [&result](std::size_t num_bytes) {
      return result->allocator.AllocateBytes(num_bytes);
    };
    // Use a dst with a null data pointer to check that the pre-packing
    // invocation doesn't write into it.
    Matrix<DstScalar> null_data_dst = result->storage_matrix.matrix;
    null_data_dst.data = nullptr;
    GlobalContext().SetRuntimeEnabledPaths(result->path);
    PrePackForMul<kAllPaths>(lhs.matrix, rhs.matrix, spec, &GlobalContext(),
                             &null_data_dst, prepacked_lhs_ptr,
                             prepacked_rhs_ptr, alloc_fn);
    RUY_CHECK(GlobalContext().last_taken_path == result->path);
  }

  life_stage = LifeStage::kHasPrepackedMatrices;
}

template <typename LhsScalar, typename RhsScalar, typename SpecType>
void TestSet<LhsScalar, RhsScalar, SpecType>::MakeResultPaths() {