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

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)

Parameters:

output_dir (Path)

Return type:

None