Skip to content
Projects
Groups
Snippets
Help
This project
Loading...
Sign in / Register
Toggle navigation
T
tic
Overview
Overview
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
wenyuanbo
tic
Commits
50f4c1d0
Commit
50f4c1d0
authored
Jun 27, 2019
by
Li
Committed by
Tianqi Chen
Jun 27, 2019
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Fix Windows build (#3429)
parent
11441c13
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
7 additions
and
18 deletions
+7
-18
include/tvm/relay/expr_functor.h
+0
-11
include/tvm/relay/pass.h
+6
-6
src/runtime/vm/vm.cc
+1
-1
No files found.
include/tvm/relay/expr_functor.h
View file @
50f4c1d0
...
...
@@ -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_
include/tvm/relay/pass.h
View file @
50f4c1d0
...
...
@@ -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
);
...
...
src/runtime/vm/vm.cc
View file @
50f4c1d0
...
...
@@ -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
(
uint
32_t
i
=
0
;
i
<
instr
.
alloc_tensor
.
ndim
;
++
i
)
{
shape
[
i
]
=
instr
.
alloc_tensor
.
shape
[
i
];
}
auto
allocator
=
MemoryManager
::
Global
()
->
GetAllocator
(
ctxs
[
0
]);
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment