diff options
| -rwxr-xr-x | analysis0510.py | 15 | ||||
| -rw-r--r-- | expt_0510.cu | 1 |
2 files changed, 12 insertions, 4 deletions
diff --git a/analysis0510.py b/analysis0510.py index c1464ce..59cc65c 100755 --- a/analysis0510.py +++ b/analysis0510.py @@ -30,7 +30,7 @@ def get_array_summary(sz: list[float]) -> tuple[str, str]: return f'{middle:.6f}', f'{avg:.6f}' -def read(file: str, need_calc: bool = False) -> None: +def read(file: str, need_calc: bool = False, output_latex: bool = False) -> None: items = {} with open(file) as fin: while s := fin.readline(): @@ -50,7 +50,13 @@ def read(file: str, need_calc: bool = False) -> None: new_result[key][x].append(element[x]) print(*keys) for key, value in new_result.items(): - print(key, *list(get_array_summary(value[x]) for x in keys)) + column = list(get_array_summary(value[x]) for x in keys) + if output_latex: + column = list(map(str, column)) + column.insert(0, str(key)) + print('\\hline\n', ' & '.join(column), '\\\\') + else: + print(key, *column) return print(items) @@ -58,7 +64,8 @@ def read(file: str, need_calc: bool = False) -> None: if __name__ == '__main__': arg_ = argparse.ArgumentParser() arg_.add_argument("file") + arg_.add_argument('--latex', action='store_true') sub_ = arg_.add_subparsers(title="sub", dest='sub') - sub_.add_parser('calc') + calc_ = sub_.add_parser('calc') parser_ = arg_.parse_args() - read(parser_.file, parser_.sub == 'calc') + read(parser_.file, parser_.sub == 'calc', parser_.latex) diff --git a/expt_0510.cu b/expt_0510.cu index 1b95385..7616c13 100644 --- a/expt_0510.cu +++ b/expt_0510.cu @@ -120,6 +120,7 @@ void initCustomSampleCPU() { } void run_kernel(size_t test_size, bool custom = true) { + // need description dim3 grid_dim = 32, block_dim = 32; unsigned long step = test_size / (grid_dim.x * block_dim.x); auto scale = 2; |