symforce.benchmarks.matrix_multiplication.generate_matrix_multiplication_benchmark module

get_matrices()

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, i)

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) –

• i (int) –

Return type:

None

generate(output_dir)

Parameters:

output_dir (Path) –

Return type:

None