From 72df483da7e96590bb88c00671c18a15bfa7eb82 Mon Sep 17 00:00:00 2001
From: Leonard Lausen <lausen@amazon.com>
Date: Wed, 5 Feb 2020 22:14:00 +0000
Subject: [PATCH] Fix lint and unrelated compiler warning

---
 include/mxnet/imperative.h   | 12 ++++++------
 src/imperative/cached_op.h   |  2 +-
 src/imperative/imperative.cc |  9 +++++----
 3 files changed, 12 insertions(+), 11 deletions(-)

diff --git a/include/mxnet/imperative.h b/include/mxnet/imperative.h
index b5545070de34..bb2122f23f99 100644
--- a/include/mxnet/imperative.h
+++ b/include/mxnet/imperative.h
@@ -92,11 +92,11 @@ class Imperative {
 
   /*! \brief DCInfo datastructure to enable deferred computation */
   class DCInfo {
-  public:
-    explicit DCInfo() {
+   public:
+    DCInfo() {
       // Default constructor provided for the sake of any.h. Should not be used.
       throw std::invalid_argument("Unsupported default constructor");
-    };
+    }
     explicit DCInfo(const std::vector<NDArray *> &inputs,
                     const std::vector<NDArray *> &outputs);
 
@@ -121,7 +121,7 @@ class Imperative {
                           const std::vector<NDArray *> &inputs,
                           const std::vector<NDArray *> &outputs);
 
-  private:
+   private:
     friend class Imperative;
 
     /*! \brief Copies of input NDArrays
@@ -225,8 +225,8 @@ class Imperative {
                 std::vector<bool>* p_save_outputs = nullptr);
   /*! \brief to record operator, return corresponding node. */
   void RecordDeferredCompute(nnvm::NodeAttrs&& attrs,
-                             std::vector<NDArray*>& inputs,
-                             std::vector<NDArray*>& outputs);
+                             const std::vector<NDArray*>& inputs,
+                             const std::vector<NDArray*>& outputs);
   /*! \brief obtain symbol representation of deferred compute session. */
   nnvm::Symbol *GetDeferredComputeSymbol(
       const std::vector<std::pair<NDArray *, std::string>> &inputs,
diff --git a/src/imperative/cached_op.h b/src/imperative/cached_op.h
index c56d8cf198b5..37cb5fe32873 100644
--- a/src/imperative/cached_op.h
+++ b/src/imperative/cached_op.h
@@ -294,7 +294,7 @@ void SetInputIndices(const nnvm::Graph& fwd_graph,
   const auto& indexed_graph = fwd_graph.indexed_graph();
   if (data_indices->ndim() || param_indices.ndim()) {
     CHECK_EQ(data_indices->ndim() + param_indices.ndim(),
-             indexed_graph.input_nodes().size());
+             static_cast<const int>(indexed_graph.input_nodes().size()));
   } else {
     std::vector<uint32_t> tmp;
     tmp.reserve(indexed_graph.input_nodes().size());
diff --git a/src/imperative/imperative.cc b/src/imperative/imperative.cc
index c1b9f2f9a5f5..77ba7866e06b 100644
--- a/src/imperative/imperative.cc
+++ b/src/imperative/imperative.cc
@@ -266,7 +266,8 @@ void Imperative::RecordOp(
       }
       inputs[i]->autograd_entry_ = std::move(entry);  // assign last to prevent cyclic reference
     } else if (save_inputs[i]) {
-      AGInfo::Get(inputs[i]->autograd_entry_.node).outputs[inputs[i]->autograd_entry_.index] = inputs[i]->Detach();
+      nnvm::NodeEntry& entry = inputs[i]->autograd_entry_;
+      AGInfo::Get(entry.node).outputs[entry.index] = inputs[i]->Detach();
     }
     node->inputs[i] = inputs[i]->autograd_entry_;
   }
@@ -292,8 +293,8 @@ void Imperative::RecordOp(
 }
 
 void Imperative::RecordDeferredCompute(nnvm::NodeAttrs &&attrs,
-                                       std::vector<NDArray *> &inputs,
-                                       std::vector<NDArray *> &outputs) {
+                                       const std::vector<NDArray *> &inputs,
+                                       const std::vector<NDArray *> &outputs) {
   CHECK(!is_recording())
       << "Autograd recording is not supported during deferred compute mode.";
 
@@ -346,7 +347,7 @@ nnvm::Symbol *Imperative::GetDeferredComputeSymbol(
   std::unordered_set<const NDArray *> missing_inputs;
   auto add_symbol_variables = [&inputs, &ndinput_to_variable,
                                &missing_inputs](const nnvm::ObjectPtr &node) {
-    if(node == nullptr) {
+    if (node == nullptr) {
       // This (nonexistant) "Node" belongs to an array created outside of deferred compute scope.
       return;
     }