| 102 | } |
| 103 | |
| 104 | int main(int argc, char* argv[]) { |
| 105 | int device = argc > 1 ? atoi(argv[1]) : 0; |
| 106 | bool console = argc > 2 ? argv[2][0] == '-' : false; |
| 107 | try { |
| 108 | af::setDevice(device); |
| 109 | af::info(); |
| 110 | printf("Simulation of shallow water equations\n"); |
| 111 | swe(console); |
| 112 | } catch (af::exception& e) { |
| 113 | fprintf(stderr, "%s\n", e.what()); |
| 114 | throw; |
| 115 | } |
| 116 | |
| 117 | return 0; |
| 118 | } |