symforce.benchmarks.matrix_multiplication.generate_matrix_multiplication_benchmark module

get_matrices()[source]

Load the matrices in the matrices folder from disk, and return their names, paths to their matrix-market files, and the matrices themselves

Filters to the first N_MATRICES results, sorted by number of nonzeros

Return type:

List[Tuple[str, Path, csr_matrix]]

generate_matrix(output_dir, matrix_name, matrix, symforce_result_is_sparse, runs_multiplier_index)[source]

Generate functions for the given matrix sparsity pattern to compute A, B, and A^T B, in sparse and dense forms

Parameters:

  • output_dir (Path) –

  • matrix_name (str) –

  • matrix (csr_matrix) –

  • symforce_result_is_sparse (bool) –

  • runs_multiplier_index (int) –

Return type:

None

generate(output_dir)[source]
Parameters:

output_dir (Path) –

Return type:

None