Commit 50f4c1d0 by Li Committed by Tianqi Chen

Fix Windows build (#3429)

parent 11441c13
......@@ -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_
......@@ -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);
......
......@@ -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]);
......
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