diff options
author | David Gross <dgross@google.com> | 2017-09-28 09:18:51 -0700 |
---|---|---|
committer | David Gross <dgross@google.com> | 2017-09-28 11:46:52 -0700 |
commit | 1f4381539b7e89c42336ee7cd1addb9a4c317b34 (patch) | |
tree | 30b62a274ed5cb72ae7725a1e155b89d6d2421ad /nn/runtime/Manager.cpp | |
parent | c7e8396248b9f42a5cec48b4e7c1c03f3ba081b4 (diff) | |
download | ml-1f4381539b7e89c42336ee7cd1addb9a4c317b34.tar.gz |
Finish implementing single-partition graphs.
This includes wiring up the partitioning algorithm
to the NN API compilation and execution phases under
debug.nn.partition.test=2: When we have a single-
partition graph, we rely on the device selection and
compilation performed by the partitioning algorithm,
rather than doing it at execution time as we were
before.
Bug: 63905942
Test: mma (user)
mma (userdebug)
ml/nn/runtime/tests (userdebug)
(with debug.nn.partition.test 0, 1, and 2;
no new failures, logcat looks plausible,
compilation happens at the expected times)
Change-Id: I6cd8b53625588d5aca81c87cc8600735e384b8e0
Diffstat (limited to 'nn/runtime/Manager.cpp')
-rw-r--r-- | nn/runtime/Manager.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/nn/runtime/Manager.cpp b/nn/runtime/Manager.cpp index e0ff32829..0980262b6 100644 --- a/nn/runtime/Manager.cpp +++ b/nn/runtime/Manager.cpp @@ -133,6 +133,9 @@ void DeviceManager::findAvailableDevices() { DeviceManager::DeviceManager() { LOG(VERBOSE) << "DeviceManager::DeviceManager"; findAvailableDevices(); +#ifdef NN_DEBUGGABLE + mPartitioning = getProp("debug.nn.partition.test"); +#endif // NN_DEBUGGABLE } } // namespace nn |