diff options
Diffstat (limited to 'src/lib')
| -rw-r--r-- | src/lib/grok/util.cc | 5 | ||||
| -rw-r--r-- | src/lib/grok/util.h | 2 |
2 files changed, 5 insertions, 2 deletions
diff --git a/src/lib/grok/util.cc b/src/lib/grok/util.cc index 8a6d2c4f9..3cbc55678 100644 --- a/src/lib/grok/util.cc +++ b/src/lib/grok/util.cc @@ -20,6 +20,7 @@ #include "util.h" +#include "../config.h" #include <boost/process.hpp> #include <future> @@ -29,10 +30,12 @@ using std::vector; vector<string> -get_gpu_names(boost::filesystem::path binary) +get_gpu_names() { namespace bp = boost::process; + auto binary = Config::instance()->grok().binary_location / "gpu_lister"; + bp::ipstream stream; bp::child child(binary, bp::std_out > stream); diff --git a/src/lib/grok/util.h b/src/lib/grok/util.h index 9996fa0e9..a78ecabca 100644 --- a/src/lib/grok/util.h +++ b/src/lib/grok/util.h @@ -24,4 +24,4 @@ #include <vector> -extern std::vector<std::string> get_gpu_names(boost::filesystem::path binary); +extern std::vector<std::string> get_gpu_names(); |
