diff options
| author | Adrian Kummerlaender | 2019-10-21 22:33:13 +0200 |
|---|---|---|
| committer | Adrian Kummerlaender | 2019-10-21 22:33:13 +0200 |
| commit | 8fb7705428a6524a436a5dd8806ad8076b7f4e08 (patch) | |
| tree | b5256587d5e5d4fc4d94799544ca0b9547ae1647 | |
| parent | 8f88aea2c963ebe63db83480fc9a4b48451220af (diff) | |
| download | boltzgen-8fb7705428a6524a436a5dd8806ad8076b7f4e08.tar boltzgen-8fb7705428a6524a436a5dd8806ad8076b7f4e08.tar.gz boltzgen-8fb7705428a6524a436a5dd8806ad8076b7f4e08.tar.bz2 boltzgen-8fb7705428a6524a436a5dd8806ad8076b7f4e08.tar.lz boltzgen-8fb7705428a6524a436a5dd8806ad8076b7f4e08.tar.xz boltzgen-8fb7705428a6524a436a5dd8806ad8076b7f4e08.tar.zst boltzgen-8fb7705428a6524a436a5dd8806ad8076b7f4e08.zip | |
Add basic argument parsing
| -rw-r--r-- | boltzgen.py | 24 | ||||
| -rw-r--r-- | test.py | 16 |
2 files changed, 24 insertions, 16 deletions
diff --git a/boltzgen.py b/boltzgen.py new file mode 100644 index 0000000..6d410c3 --- /dev/null +++ b/boltzgen.py @@ -0,0 +1,24 @@ +import argparse +import string + +from boltzgen import * + +argparser = argparse.ArgumentParser(description='Generate LBM kernels in various languages using a symbolic description.') +argparser.add_argument('language', help = 'Target language (currently either "opencl" or "cpp")') + +args = argparser.parse_args() + +lbm = LBM(D2Q9) +geometry = Geometry(32,32) + +src = source( + args.language, + D2Q9, + lbm.moments(), + lbm.bgk(f_eq = lbm.equilibrium(), tau = 0.6), + "", + 'float', + geometry +) + +print(src) diff --git a/test.py b/test.py deleted file mode 100644 index 0eec050..0000000 --- a/test.py +++ /dev/null @@ -1,16 +0,0 @@ -from boltzgen import * - -lbm = LBM(D2Q9) -geometry = Geometry(32,32) - -src = source( - 'opencl', - D2Q9, - lbm.moments(), - lbm.bgk(f_eq = lbm.equilibrium(), tau = 0.6), - "", - 'float', - geometry -) - -print(src) |