(df)
| 372 | # df = results.matmul_results_square() |
| 373 | |
| 374 | def clean_df(df): |
| 375 | df = df.loc[df['algo'].isin(ALGOS)] |
| 376 | non_encode_algos = ['Bolt 8B', 'Bolt 16B', 'Bolt 32B'] |
| 377 | # rm_idxs = (df['algo'] == 'Bolt 32B') * (df['enc'] == 1) |
| 378 | rm_idxs = (df['algo'].isin(non_encode_algos)) * (df['enc'] == 1) |
| 379 | df = df.loc[~rm_idxs] |
| 380 | |
| 381 | df['algo'].loc[df['algo'] == 'Floats'] = 'Floats (BLAS)' |
| 382 | return df |
| 383 | |
| 384 | df = results.matmul_results(which='square') |
| 385 | df = clean_df(df) |