Skip to content

Commit

Permalink
Fix Windows build (apache#3429)
Browse files Browse the repository at this point in the history
  • Loading branch information
aksarben09 authored and wweic committed Jun 30, 2019
1 parent 6382276 commit ab6e6bf
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 18 deletions.
11 changes: 0 additions & 11 deletions include/tvm/relay/expr_functor.h
Original file line number Diff line number Diff line change
Expand Up @@ -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_
12 changes: 6 additions & 6 deletions include/tvm/relay/pass.h
Original file line number Diff line number Diff line change
Expand Up @@ -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);

Expand Down
2 changes: 1 addition & 1 deletion src/runtime/vm/vm.cc
Original file line number Diff line number Diff line change
Expand Up @@ -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]);
Expand Down

0 comments on commit ab6e6bf

Please sign in to comment.