| 164 | } |
| 165 | |
| 166 | int main(int argc, char **argv) { |
| 167 | int device = argc > 1 ? atoi(argv[1]) : 0; |
| 168 | bool console = argc > 2 ? argv[2][0] == '-' : false; |
| 169 | int k = argc > 3 ? atoi(argv[3]) : 8; |
| 170 | |
| 171 | try { |
| 172 | af::setDevice(device); |
| 173 | af::info(); |
| 174 | return kmeans_demo(k, console); |
| 175 | |
| 176 | } catch (af::exception &ae) { std::cerr << ae.what() << std::endl; } |
| 177 | |
| 178 | return 0; |
| 179 | } |
nothing calls this directly
no test coverage detected