From 44d2419eb4caace1bafdea2c4800379ba48d13f4 Mon Sep 17 00:00:00 2001
From: =?utf8?q?=EC=9D=B4=ED=95=9C=EC=A2=85/=EB=8F=99=EC=9E=91=EC=A0=9C?=
 =?utf8?q?=EC=96=B4Lab=28SR=29/Engineer/=EC=82=BC=EC=84=B1=EC=A0=84?=
 =?utf8?q?=EC=9E=90?= <hanjoung.lee@samsung.com>
Date: Thu, 22 Mar 2018 18:34:28 +0900
Subject: [PATCH] Enable ModelBuilder::createCompilation (#156)

Now that CompilationBuilder is available we can enable it.
---
 src/runtime/ref/nn/runtime/ModelBuilder.cpp | 14 ++++++--------
 src/runtime/ref/nn/runtime/ModelBuilder.h   |  6 +++---
 2 files changed, 9 insertions(+), 11 deletions(-)

diff --git a/src/runtime/ref/nn/runtime/ModelBuilder.cpp b/src/runtime/ref/nn/runtime/ModelBuilder.cpp
index 3794b26..c3d25f2 100644
--- a/src/runtime/ref/nn/runtime/ModelBuilder.cpp
+++ b/src/runtime/ref/nn/runtime/ModelBuilder.cpp
@@ -17,10 +17,11 @@
 
 #include "ModelBuilder.h"
 
+#include "CompilationBuilder.h"
+
 #if 0
 #define LOG_TAG "ModelBuilder"
 
-#include "CompilationBuilder.h"
 #include "Utils.h"
 
 #include <map>
@@ -293,21 +294,18 @@ int ModelBuilder::identifyInputsAndOutputs(uint32_t inputCount, const uint32_t*
     return ANEURALNETWORKS_NO_ERROR;
 #endif
 }
-#if 0 // CompilationBuilder is needed.
+
 int ModelBuilder::createCompilation(CompilationBuilder** compilation) {
-    // Dummy Implementation
-    return 0;
-#if 0
     if (!mCompletedModel) {
-        LOG(ERROR) << "ANeuralNetworksCompilation_create passed an unfinished model";
+        // TODO-NNRT Uncomment once LOG is available
+        // LOG(ERROR) << "ANeuralNetworksCompilation_create passed an unfinished model";
         *compilation = nullptr;
         return ANEURALNETWORKS_BAD_STATE;
     }
     *compilation = new CompilationBuilder(this);
     return (*compilation ? ANEURALNETWORKS_NO_ERROR : ANEURALNETWORKS_OUT_OF_MEMORY);
-#endif
 }
-#endif
+
 int ModelBuilder::finish() {
     // Dummy Implementation
     return 0;
diff --git a/src/runtime/ref/nn/runtime/ModelBuilder.h b/src/runtime/ref/nn/runtime/ModelBuilder.h
index 9a6bcbe..5ecee4b 100644
--- a/src/runtime/ref/nn/runtime/ModelBuilder.h
+++ b/src/runtime/ref/nn/runtime/ModelBuilder.h
@@ -33,8 +33,10 @@
 #endif
 namespace android {
 namespace nn {
-#if 0
+
 class CompilationBuilder;
+
+#if 0
 class Device;
 class ExecutionPlan;
 class ExecutionStep;
@@ -57,9 +59,7 @@ public:
 
     int finish();
     bool isFinished() const { return mCompletedModel; }
-#if 0 // CompilationBuilder is needed.
     int createCompilation(CompilationBuilder** compilation);
-#endif
 #if 0 // NeuralNetworksWrapper.h is needed.
     void setHidlModel(Model* model) const;
 #endif
-- 
2.7.4