Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

[MXNET-432] Add Foreach #11531

Merged
merged 135 commits into from
Jul 2, 2018
Merged
Changes from 1 commit
Commits
Show all changes
135 commits
Select commit Hold shift + click to select a range
e86d92c
Test input a graph.
zheng-da Mar 20, 2018
7eacfab
Update foreach to execute the subgraph.
zheng-da Mar 21, 2018
6874f7d
print inputs/outputs in foreach.
zheng-da Mar 21, 2018
8f5e62e
Remove print.
zheng-da Mar 23, 2018
1dd35a8
add test code for foreach.
zheng-da Mar 23, 2018
1f117cb
exec foreach outside the engine.
zheng-da Apr 5, 2018
cc29fc1
Implements forward of foreach.
zheng-da Apr 5, 2018
036aada
Add support for variable numbers of inputs and outputs.
zheng-da Apr 6, 2018
84e0e24
Add a python wrapper for foreach.
zheng-da Apr 6, 2018
f2a28f0
Fix the order of inputs.
zheng-da Apr 6, 2018
1c4cf0e
hide C version of foreach.
zheng-da Apr 7, 2018
9aa896d
fix a bug temporarily.
zheng-da Apr 7, 2018
6784408
add test with lstm.
zheng-da Apr 6, 2018
f488647
Test free variables.
zheng-da Apr 7, 2018
d9b0c50
change for the new interface of InputGraph attribute.
zheng-da Apr 9, 2018
1188d97
Add attribute to the subgraph.
zheng-da Apr 11, 2018
c0cd6ac
Handle free variables.
zheng-da Apr 11, 2018
74d280b
Get all input symbols of a subgraph.
zheng-da Apr 13, 2018
2bc80e3
Fix shape, dtype and storage inference.
zheng-da Apr 13, 2018
68faa17
reorganize the output of foreach.
zheng-da Apr 14, 2018
3751ca7
Add a gluon RNN unroll with symbol foreach.
zheng-da Apr 14, 2018
b98e06d
print unnecessary print.
zheng-da Apr 16, 2018
fc575fe
have imperative and symbolic foreach.
zheng-da Apr 16, 2018
37da6fb
Fix an error after moving foreach.
zheng-da Apr 18, 2018
f41235c
Fix imperative foreach
zheng-da Apr 18, 2018
214c1c2
Fix a minor problem.
zheng-da Apr 24, 2018
9aabc74
Use CachedOp to execute subgraph.
zheng-da Apr 30, 2018
7fc0155
update TODO.
zheng-da May 1, 2018
0d3613a
make foreach op use FStatefulComputeEx.
zheng-da May 1, 2018
f33d0f4
Add backward.
zheng-da May 2, 2018
d82dd30
Fix bugs.
zheng-da May 4, 2018
868c9f2
enable backward test in lstm.
zheng-da May 4, 2018
84e1877
Fix a bug in foreach backward for free variables.
zheng-da May 7, 2018
00b8b1c
change for the new CachedOp.
zheng-da May 9, 2018
f2e324e
Detect the backward computation.
zheng-da May 9, 2018
e1322d1
Fix bugs in foreach.
zheng-da May 9, 2018
98955a4
fix tests.
zheng-da May 10, 2018
d8c9b1f
update tests.
zheng-da May 11, 2018
32e3b17
check state shape.
zheng-da May 12, 2018
8caf708
enable nested foreach.
zheng-da May 14, 2018
d4ef381
remove print.
zheng-da May 16, 2018
4270032
fix a bug in test.
zheng-da May 17, 2018
b54f234
handle infer storage type for backward.
zheng-da May 18, 2018
14d319b
address comments.
zheng-da May 18, 2018
0e666a9
address comments.
zheng-da May 18, 2018
255c478
move some common functions out.
zheng-da May 18, 2018
2beb3f3
address comments.
zheng-da May 18, 2018
716bc6a
fix lint.
zheng-da May 18, 2018
dd5f862
Fix lint.
zheng-da May 18, 2018
b60157a
add doc.
zheng-da May 19, 2018
57b2ba5
undo modification in imperative.h
zheng-da May 19, 2018
7c49057
add doc and remove example code.
zheng-da May 19, 2018
1045908
fix lint.
zheng-da May 19, 2018
e8ec3aa
fix lint.
zheng-da May 19, 2018
e4f5808
Fix lint.
zheng-da May 19, 2018
c078cbf
make nd.foreach and sym.foreach consistent.
zheng-da May 21, 2018
b965e6e
fix compile error.
zheng-da May 21, 2018
57fcb84
address comments.
zheng-da May 21, 2018
c03c56f
update.
zheng-da May 21, 2018
224f3e2
check for loop only works for dense arrays.
zheng-da May 22, 2018
cd67c6f
move control flow op out of nn/
zheng-da May 22, 2018
742ef40
fix include.
zheng-da May 22, 2018
4492949
add a test in gluon.
zheng-da May 22, 2018
26e3e7e
small fix.
zheng-da May 22, 2018
6bff448
remove subgraph_name
zheng-da May 22, 2018
1e4cd45
create loop state for reuse in the future.
zheng-da May 22, 2018
7079e73
work for GPU.
zheng-da May 22, 2018
64f4362
Fix tests.
zheng-da May 29, 2018
31d9112
Fix bugs caused by ctypes (#29)
junrushao1994 May 30, 2018
601edbe
Add save/load json in testcases for foreach (#30)
junrushao1994 Jun 1, 2018
f4da935
support subgraph in stateful executor.
zheng-da Jun 4, 2018
90b7829
Fix compilation.
zheng-da Jun 4, 2018
ae3ea22
move code.
zheng-da May 22, 2018
0db16f0
Revert "remove subgraph_name"
zheng-da May 23, 2018
5f626ae
cut graph.
zheng-da May 25, 2018
f2c428f
rename new var nodes.
zheng-da May 26, 2018
2a69257
fix a bug when a subgraph has variable nodes.
zheng-da Jun 8, 2018
efeedd6
Fix a bug of getting symbols.
zheng-da Jun 8, 2018
28fe469
copy var nodes.
zheng-da Jun 8, 2018
cf91c59
Fix getting op states.
zheng-da Jun 13, 2018
f2edf2a
fix lint error.
zheng-da Jun 13, 2018
a35899f
address comments.
zheng-da Jun 13, 2018
8c6aca0
fix lint error.
zheng-da Jun 13, 2018
ccaf388
simplify the execution of subgraph in the main thread.
zheng-da Jun 13, 2018
25cf8ac
fix lint error.
zheng-da Jun 13, 2018
51de14c
avoid waiting for computation in each iteration.
zheng-da Jun 13, 2018
3eb0bc1
reuse cached op for inference.
zheng-da Jun 14, 2018
4a0ff21
share memory across mini-batches.
zheng-da Jun 14, 2018
8766cb2
reuse memory.
zheng-da Jun 14, 2018
97b9074
add tests for multiple batches.
zheng-da Jun 14, 2018
e38b7f4
remove entry.
zheng-da Jun 15, 2018
198bcfb
add benchmark for foreach.
zheng-da Jun 16, 2018
811acb3
benchmark large batch size.
zheng-da Jun 16, 2018
24fa83b
Fix the benchmark for GPU.
zheng-da Jun 17, 2018
550e48a
address comments.
zheng-da Jun 17, 2018
97d0332
update shape/dtype/storage inference.
zheng-da Jun 17, 2018
0b0a36e
update contrib API docs.
zheng-da Jun 17, 2018
f1ff55d
support nested foreach.
zheng-da Jun 18, 2018
156f1c8
use a single CachedOp for all iterations.
zheng-da Jun 19, 2018
871fd3b
use large dim.
zheng-da Jun 19, 2018
b5dfc3f
update benchmark.
zheng-da Jun 19, 2018
202a74c
update benchmark.
zheng-da Jun 19, 2018
0606c3c
update benchmark.
zheng-da Jun 19, 2018
6019de5
update benchmark.
zheng-da Jun 19, 2018
045186d
return symbol arrays correctly in MXSymbolCutSubgraph.
zheng-da Jun 20, 2018
484309e
return symbol arrays in MXSymbolGetInputSymbols.
zheng-da Jun 20, 2018
0ebd5e5
fix lint error.
zheng-da Jun 20, 2018
a9e253d
use cachedop to infer storage in backward.
zheng-da Jun 21, 2018
1f8469f
fix scala API.
zheng-da Jun 21, 2018
25e15a0
update comments.
zheng-da Jun 21, 2018
ff4eea0
fix scala.
zheng-da Jun 21, 2018
b8aa62a
fix test.
zheng-da Jun 21, 2018
64e4ff6
fix attribute name.
zheng-da Jun 21, 2018
d243c12
move benchmark.
zheng-da Jun 21, 2018
3afc4d4
fix the mapping of operator inputs/outputs and subgraph inputs/outputs.
zheng-da Jun 22, 2018
62901fe
add tests for dtype/shape inference.
zheng-da Jun 23, 2018
14b8fb9
reorganize tests.
zheng-da Jun 23, 2018
f7d7f17
fix a bug of cutting NodeEntry.
zheng-da Jun 23, 2018
7d012d9
fix lint error.
zheng-da Jun 23, 2018
b83253d
handle the case that outputs are inputs.
zheng-da Jun 24, 2018
275bbf1
handle the case that inputs aren't used.
zheng-da Jun 24, 2018
0e6df9a
handle the case without output data.
zheng-da Jun 25, 2018
dfadc8d
fix a bug in foreach backward.
zheng-da Jun 25, 2018
5e9cf5f
fix a bug when there isn't output data.
zheng-da Jun 25, 2018
696f53c
Fix lint error.
zheng-da Jun 26, 2018
094977d
test diff Gluon RNN cells.
zheng-da Jun 26, 2018
7b016ae
test all symbol RNN cells.
zheng-da Jun 26, 2018
9609ce8
adjust the test precision.
zheng-da Jun 26, 2018
fa8abbd
Fix a bug in getting a list of variable names.
zheng-da Jun 26, 2018
8e74d80
fix lint error.
zheng-da Jun 26, 2018
2439105
Test 1D array.
zheng-da Jun 27, 2018
53cdbfa
fix a bug when subgraph inputs and outputs share NDArray.
zheng-da Jun 27, 2018
9bff317
fix.
zheng-da Jun 28, 2018
d3687ef
fix
zheng-da Jun 28, 2018
392a7e4
add comments.
zheng-da Jul 2, 2018
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fix lint error.
zheng-da committed Jul 2, 2018

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
commit 0ebd5e5fed976711a0c47714fd14efb0eb94708f
2 changes: 1 addition & 1 deletion python/mxnet/symbol/contrib.py
Original file line number Diff line number Diff line change
@@ -30,7 +30,7 @@
pass

from . import symbol
from ..base import _LIB, c_array, check_call
from ..base import _LIB, check_call
from ..base import SymbolHandle, _as_list
from ..attribute import AttrScope