Make tools/ for core binaries, stow scripts/ in tools/extra

Collect core Caffe tools like train_net, device_query, etc. together in
tools/ and include helper scripts under tools/extra.

1 files changed, 23 insertions, 0 deletions

diff --git a/tools/device_query.cpp b/tools/device_query.cpp
new file mode 100644
index 00000000..88bf5aaa
--- /dev/null
+++ b/tools/device_query.cpp
@@ -0,0 +1,23 @@
+// Copyright 2014 Sergio Guadarrama
+
+
+#include "caffe/common.hpp"
+#include "caffe/net.hpp"
+
+
+using namespace caffe;
+
+int main(int argc, char** argv) {
+  if (argc > 2) {
+    LOG(ERROR) << "device_query [device_id=0]";
+    return 0;
+  }
+  if (argc == 2) {
+    LOG(INFO) << "Querying device_id=" << argv[1];
+    Caffe::SetDevice(atoi(argv[1]));
+    Caffe::DeviceQuery();
+  } else {
+    Caffe::DeviceQuery();
+  }
+  return 0;
+}