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

dynamic custom operator support #15921

Merged
merged 139 commits into from
Dec 6, 2019
Merged

dynamic custom operator support #15921

Changes from 1 commit
Commits
Show all changes
139 commits
Select commit Hold shift + click to select a range
5030a65
fixed example to use absolute path
Aug 15, 2019
23a226a
added example for custom ops, added support for custom op registration
Aug 16, 2019
67c22c0
added fcompute registration for loaded operators
Aug 17, 2019
915c1d5
changed dynamic ops to be contrib
Aug 17, 2019
f568e3d
added num in/out
Aug 18, 2019
8e12588
removed contrib op registration
Aug 20, 2019
1e27a47
added support for infer shape, updated example to call operator
Aug 20, 2019
9aecf86
fixed whitespace
Aug 20, 2019
02deacf
fixed whitespace
Aug 20, 2019
cf9350d
fixed whitespace
Aug 20, 2019
ada3895
Merge branch 'master' of https://github.com/apache/incubator-mxnet in…
Aug 23, 2019
38e77a5
added temporary support for operator multi-registration
Aug 23, 2019
7b8f6a2
insanity checked
Aug 23, 2019
5b817bd
update docblocks
rondogency Aug 23, 2019
3bccfbe
small format fix
rondogency Aug 23, 2019
a8c19c8
fix unittest with correct library
rondogency Aug 23, 2019
2f34471
implement InferType
rondogency Aug 27, 2019
3502aa9
Merge branch 'master' of https://github.com/apache/incubator-mxnet in…
Aug 27, 2019
52e687b
Merge branch 'dynamic_op' of https://github.com/samskalicky/incubator…
Aug 27, 2019
5438a35
Merge branch 'master' of https://github.com/apache/incubator-mxnet in…
Aug 27, 2019
592249a
initial support for resource manager, temp space
Aug 27, 2019
3186d60
fixed formatting
Aug 27, 2019
e8b413b
Merge branch 'master' of https://github.com/apache/incubator-mxnet in…
Aug 29, 2019
bf549b4
changed decltype to typedef
Aug 29, 2019
439ee20
fixed whitespace
Aug 29, 2019
bba25db
Added windows declaration types, change APIs to return MXReturnValue …
Aug 29, 2019
a681f61
added library version number, API to get, and check to validate
Aug 29, 2019
711f9a3
Changed CMakeLists to build lib_ops instead of lib_api, updated lib_a…
Aug 29, 2019
172129f
Merge branch 'master' of https://github.com/apache/incubator-mxnet in…
Aug 29, 2019
5af1736
add prototype of subgraph op
rondogency Aug 29, 2019
33d9cd7
implement FMutateInput as optional attribute
rondogency Aug 30, 2019
4576570
fix sanity check
rondogency Aug 30, 2019
6f3e3d9
replace fcompute to fcomputeEx and implement simple finferstoragetype
rondogency Sep 3, 2019
9587483
Merge branch 'master' of https://github.com/apache/incubator-mxnet in…
Sep 4, 2019
34efb2b
Merge branch 'dynamic_op' of https://github.com/samskalicky/incubator…
Sep 4, 2019
ff9a868
changed fcompute to forward
Sep 4, 2019
0be218b
initial commit with fgradient support
Sep 4, 2019
570a059
enabled gradient registration
Sep 4, 2019
4b01932
Merge branch 'master' of https://github.com/apache/incubator-mxnet in…
Sep 4, 2019
e4be175
fixed whitespace
Sep 4, 2019
8cfcc85
fixed example to use absolute path
Aug 15, 2019
9884ec6
added example for custom ops, added support for custom op registration
Aug 16, 2019
8e21600
added fcompute registration for loaded operators
Aug 17, 2019
794e30b
changed dynamic ops to be contrib
Aug 17, 2019
8fbf664
added num in/out
Aug 18, 2019
e7c6e8f
removed contrib op registration
Aug 20, 2019
6047378
added support for infer shape, updated example to call operator
Aug 20, 2019
d1587ab
fixed whitespace
Aug 20, 2019
0ee56c9
fixed whitespace
Aug 20, 2019
adc9770
fixed whitespace
Aug 20, 2019
5c06d47
added temporary support for operator multi-registration
Aug 23, 2019
9136839
insanity checked
Aug 23, 2019
ffe7623
update docblocks
rondogency Aug 23, 2019
435e01e
small format fix
rondogency Aug 23, 2019
0de79a9
fix unittest with correct library
rondogency Aug 23, 2019
0d6f7b0
implement InferType
rondogency Aug 27, 2019
18b028e
initial support for resource manager, temp space
Aug 27, 2019
a4690b4
fixed formatting
Aug 27, 2019
c901828
changed decltype to typedef
Aug 29, 2019
5ddb919
fixed whitespace
Aug 29, 2019
7b4c4e6
Added windows declaration types, change APIs to return MXReturnValue …
Aug 29, 2019
18117ec
added library version number, API to get, and check to validate
Aug 29, 2019
ee65419
Changed CMakeLists to build lib_ops instead of lib_api, updated lib_a…
Aug 29, 2019
c66438c
add prototype of subgraph op
rondogency Aug 29, 2019
698a0b6
implement FMutateInput as optional attribute
rondogency Aug 30, 2019
bd55612
fix sanity check
rondogency Aug 30, 2019
35ff973
replace fcompute to fcomputeEx and implement simple finferstoragetype
rondogency Sep 3, 2019
f243e2f
changed fcompute to forward
Sep 4, 2019
efbb858
initial commit with fgradient support
Sep 4, 2019
0032143
enabled gradient registration
Sep 4, 2019
14ef3a7
fixed whitespace
Sep 4, 2019
eec71d6
prototype of createopstate and fstatefulcompute
rondogency Sep 6, 2019
abcb8cb
make custom state op interface work
rondogency Sep 6, 2019
9cf0455
subgraph forward
rondogency Sep 9, 2019
82f1bff
refactor stateful forward and add op resource
rondogency Sep 10, 2019
f7ff481
Merge branch 'master' of https://github.com/apache/incubator-mxnet in…
Sep 10, 2019
ba563d2
wip gemm backward
Sep 10, 2019
a9b7215
Merge branch 'dynamic_op' of https://github.com/samskalicky/incubator…
Sep 10, 2019
7bf4f7a
stateful backward and subgraph test
rondogency Sep 11, 2019
8aec7ac
implement gemm and state gemm, refactor test files
rondogency Sep 12, 2019
39e3d6b
add body to pure virtual destructor
rondogency Sep 12, 2019
b3ba028
subgraph passing from python to custom lib
rondogency Sep 23, 2019
c9d8498
Merge branch 'master' into dynamic_op
rondogency Sep 24, 2019
1686273
rm lib_api c++11 dep, rm warpctc, add rm flag
rondogency Sep 26, 2019
7009ad4
fix conflict
rondogency Sep 26, 2019
4b73179
subgraph json parsing utility
rondogency Sep 28, 2019
dca521e
add data size and fix unsigned warnings
rondogency Sep 29, 2019
baed04e
use c++ struct and fix cpplint
rondogency Sep 30, 2019
aedcf91
refactor op registry
rondogency Sep 30, 2019
75102a3
fix line length and win array of ci; condense lines
rondogency Sep 30, 2019
9c29deb
Merge remote-tracking branch 'upstream/master' into dynamic_op
rondogency Sep 30, 2019
c5a3ed6
add mxnet_extension dir
rondogency Oct 1, 2019
44683f1
Merge branch 'master' of https://github.com/apache/incubator-mxnet in…
Oct 1, 2019
d1b6c8e
Merge branch 'dynamic_op' of https://github.com/samskalicky/incubator…
Oct 1, 2019
44affc7
fixed extension to be dll for windows
Oct 1, 2019
ef1d4cf
updated examples to use the same format as the example in the top-lev…
Oct 1, 2019
24d8cc3
removed destructor for CustomStatefulOp
Oct 1, 2019
279a989
fix error in gemm test and clear up subgraph test
rondogency Oct 2, 2019
5db9e97
merge with dynamic_op
rondogency Oct 2, 2019
75b1169
lib path fix
rondogency Oct 2, 2019
79c0e3a
add unittest for custom op
rondogency Oct 2, 2019
11d3344
update Makefile revolve merge
rondogency Oct 2, 2019
de157a8
Merge branch 'master' of https://github.com/apache/incubator-mxnet in…
Oct 2, 2019
28450b5
Merge branch 'dynamic_op' of https://github.com/samskalicky/incubator…
Oct 2, 2019
9504b33
fix test and rename folder
rondogency Oct 2, 2019
cf27d57
fix makefile rename
rondogency Oct 2, 2019
7f456d4
fix cmake rename
rondogency Oct 2, 2019
e50819b
add explicit cpu context
rondogency Oct 2, 2019
5984f3a
Merge remote-tracking branch 'upstream/master' into dynamic_op
rondogency Oct 2, 2019
bd2c3a0
wkcn feedback: change mxtensor func name. use c++11 flag
rondogency Oct 3, 2019
b07e46b
add operator keyward test and refine info print
rondogency Oct 3, 2019
2466d67
using typedef in forward
rondogency Oct 3, 2019
e041400
small refine of docblock
rondogency Oct 4, 2019
f16942c
change names
rondogency Oct 8, 2019
50a6b64
add separate stateful compute and pass state_op ptr
rondogency Oct 8, 2019
adb0415
user example using opresource alloc
rondogency Oct 14, 2019
6148ef8
Merge remote-tracking branch 'upstream/master' into dynamic_op
rondogency Oct 14, 2019
6d9ac54
Merge remote-tracking branch 'upstream/master' into dynamic_op
rondogency Oct 15, 2019
7c256cd
Merge remote-tracking branch 'upstream/master' into dynamic_op
rondogency Oct 15, 2019
6e824fb
Merge branch 'master' of https://github.com/apache/incubator-mxnet in…
Oct 18, 2019
40c471b
Merge branch 'dynamic_op' of https://github.com/samskalicky/incubator…
Oct 18, 2019
dfb5946
added DLTensor into MXTensor
Oct 18, 2019
5146fd5
fixed whitespace
Oct 18, 2019
1b9fee2
added error check when DLTensor does not support MXNet data type
Oct 18, 2019
5761891
changed to throw runtime exception
Oct 18, 2019
ef840b4
changed include to stdexcept
Oct 18, 2019
bba61b3
retrigger CI
wkcn Oct 18, 2019
53d18ec
empty commit
Oct 18, 2019
e0c778c
Merge branch 'dynamic_op' of https://github.com/samskalicky/incubator…
Oct 18, 2019
141328f
empty commit
Oct 18, 2019
deacae2
Merge branch 'master' of https://github.com/apache/incubator-mxnet in…
Oct 18, 2019
2b2c6a4
Merge branch 'master' into dynamic_op
szha Oct 22, 2019
ed8ac16
remove merge conflict
rondogency Oct 23, 2019
56b0e28
add setdltensor for easy use and add docs
rondogency Oct 23, 2019
1bd166e
Merge branch 'master' into dynamic_op
wkcn Nov 26, 2019
50c8aea
CI
wkcn Nov 26, 2019
34a9ee9
re-trigger CI
wkcn Nov 28, 2019
9910c39
ci
wkcn Dec 5, 2019
5fd4314
ci
wkcn Dec 6, 2019
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
2 changes: 1 addition & 1 deletion Makefile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -660,7 +660,7 @@ pylint:
python3 -m pylint --rcfile=$(ROOTDIR)/ci/other/pylintrc --ignore-patterns=".*\.so$$,.*\.dll$$,.*\.dylib$$" python/mxnet tools/caffe_converter/*.py

sample_lib:
$(CXX) -shared -fPIC -std=c++11 example/lib_api/mylib.cc -o libsample_lib.so -I include/mxnet
$(CXX) -shared -fPIC -std=c++11 example/lib_ops/mylib.cc -o libsample_lib.so -I include/mxnet
samskalicky marked this conversation as resolved.
Show resolved Hide resolved

doc: docs

Expand Down