Commit 50f4c1d0 by Li Committed by Tianqi Chen

Fix Windows build (#3429)

parent 11441c13
...@@ -237,17 +237,6 @@ class ExprMutator ...@@ -237,17 +237,6 @@ class ExprMutator
*/ */
void PostOrderVisit(const Expr& node, std::function<void(const Expr&)> fvisit); 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 relay
} // namespace tvm } // namespace tvm
#endif // TVM_RELAY_EXPR_FUNCTOR_H_ #endif // TVM_RELAY_EXPR_FUNCTOR_H_
...@@ -444,14 +444,14 @@ TVM_DLL Array<Pattern> UnmatchedCases(const Match& match, const Module& mod); ...@@ -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); 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. * Parameter binding can only happen if expr is a Function.
* \param bind_map The variable to expression map that will be used to help the * binds cannot change internal arguments of internal functions.
* binding.
* *
* \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); TVM_DLL Expr Bind(const Expr& expr, const tvm::Map<Var, Expr>& bind_map);
......
...@@ -720,7 +720,7 @@ void VirtualMachine::Run() { ...@@ -720,7 +720,7 @@ void VirtualMachine::Run() {
} }
case Opcode::AllocTensor: { case Opcode::AllocTensor: {
auto shape = std::vector<int64_t>(instr.alloc_tensor.ndim); 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]; shape[i] = instr.alloc_tensor.shape[i];
} }
auto allocator = MemoryManager::Global()->GetAllocator(ctxs[0]); auto allocator = MemoryManager::Global()->GetAllocator(ctxs[0]);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment