Fix Windows build (#3429)
authorLi <ritsu1228@gmail.com>
Thu, 27 Jun 2019 16:36:10 +0000 (09:36 -0700)
committerTianqi Chen <tqchen@users.noreply.github.com>
Thu, 27 Jun 2019 16:36:10 +0000 (09:36 -0700)
include/tvm/relay/expr_functor.h
include/tvm/relay/pass.h
src/runtime/vm/vm.cc

index 27c0aa0..7e6fb7f 100644 (file)
@@ -237,17 +237,6 @@ class ExprMutator
  */
 void PostOrderVisit(const Expr& node, std::function<void(const Expr&)> fvisit);
 
-/*
- * \brief Bind function parameters or free variables.
- *
- * Parameter binding can only happen if expr is a Function.
- * binds cannot change internal arguments of internal functions.
- *
- * \param expr The function to be binded.
- * \param binds The map of arguments to
- */
-Expr Bind(const Expr& expr, const tvm::Map<Var, Expr>& binds);
-
 }  // namespace relay
 }  // namespace tvm
 #endif  // TVM_RELAY_EXPR_FUNCTOR_H_
index fff630f..294d22b 100644 (file)
@@ -444,14 +444,14 @@ TVM_DLL Array<Pattern> UnmatchedCases(const Match& match, const Module& mod);
  */
 TVM_DLL Expr PartialEval(const Expr& e, const Module& mod);
 
-/*!
- * \brief Bind the free variables to a Relay expression.
+/*
+ * \brief Bind function parameters or free variables.
  *
- * \param expr The expression.
- * \param bind_map The variable to expression map that will be used to help the
- *        binding.
+ * Parameter binding can only happen if expr is a Function.
+ * binds cannot change internal arguments of internal functions.
  *
- * \return The updated expression.
+ * \param expr The function to be binded.
+ * \param binds The map of arguments to
  */
 TVM_DLL Expr Bind(const Expr& expr, const tvm::Map<Var, Expr>& bind_map);
 
index 5ba2098..3df4047 100644 (file)
@@ -720,7 +720,7 @@ void VirtualMachine::Run() {
       }
       case Opcode::AllocTensor: {
         auto shape = std::vector<int64_t>(instr.alloc_tensor.ndim);
-        for (uint i = 0; i < instr.alloc_tensor.ndim; ++i) {
+        for (uint32_t i = 0; i < instr.alloc_tensor.ndim; ++i) {
           shape[i] = instr.alloc_tensor.shape[i];
         }
         auto allocator = MemoryManager::Global()->GetAllocator(ctxs[0]);