Megvii Engine Team
|
dc2507453b
|
feat(mge): add python custom op
GitOrigin-RevId: 35da0bb301
|
5 years ago |
Megvii Engine Team
|
14d8b709e1
|
perf(mge/imperative): add mini graph to partially replace proxy graph
GitOrigin-RevId: 73e2529ba5
|
5 years ago |
Megvii Engine Team
|
147cef52a0
|
refactor(mge/imperative): implement new tensor system
GitOrigin-RevId: 2dd4e460ac
|
5 years ago |
Megvii Engine Team
|
577366a203
|
feat(mge/imperative): basic impl of python c extension for opdef
fix(imperative): fix refcount management on cpython opdef
refactor(mge/imperative): fix compilation for python3.6
GitOrigin-RevId: 332a516895
|
5 years ago |
Megvii Engine Team
|
9a79342d35
|
refactor(mge/virtualdep): remove virtualdep opdef
GitOrigin-RevId: bbe3ae3fa3
|
5 years ago |
Megvii Engine Team
|
88b402ef62
|
feat(mge/trace): tracing use id to set priority
GitOrigin-RevId: 6e1f1ece0e
|
5 years ago |
Megvii Engine Team
|
69e3e32240
|
feat(imperative): auto generated opdef header and python binding
GitOrigin-RevId: d2f22ad5fe
|
5 years ago |
Megvii Engine Team
|
7aa54b0ec6
|
feat(mge): enable memory swap and drop/recomputation
GitOrigin-RevId: c56c87c88b
|
5 years ago |
Megvii Engine Team
|
cc85047bf0
|
fix(mge/trace): fix sublinear in trace
GitOrigin-RevId: 356dcd9523
|
5 years ago |
Megvii Engine Team
|
064c774f05
|
feat(imperative): impl hashable for SendRecv and add virtual input for Recv
GitOrigin-RevId: 5e8c27ac81
|
5 years ago |
Megvii Engine Team
|
798f7b3ef6
|
feat(imperative): add virtual_deps op
GitOrigin-RevId: 5845520989
|
5 years ago |
Megvii Engine Team
|
634de59061
|
feat(mge/imperative): add valid flag of `infer_output_attrs_fallible`
GitOrigin-RevId: b2b32774ee
|
5 years ago |
Megvii Engine Team
|
50c4daac85
|
feat(mge/interpreter): add async_level mechanism for Interpreter
GitOrigin-RevId: 8615a23b75
|
5 years ago |
Megvii Engine Team
|
120e719e17
|
refactor(imperative): refactor OpTrait methods and registration
remove unused op methods and enable OpTrait registered from seperated files, also fix cond_take's infer_output_attrs
GitOrigin-RevId: 134c8215ce
|
5 years ago |
Megvii Engine Team
|
1217801133
|
perf(mge): add opdef for broadcast
GitOrigin-RevId: 92f0af29eb
|
5 years ago |
Megvii Engine Team
|
fccb2510c4
|
perf(mge): add opdef for elemwise and batchnorm
GitOrigin-RevId: d51fad9867
|
5 years ago |
Megvii Engine Team
|
e8bf5bc002
|
fix(mge): ensure contiguous when passing tensor from graph rt to imperative rt
GitOrigin-RevId: e4d944343d
|
5 years ago |
Megvii Engine Team
|
dacc485407
|
fix(build/windows/macos): fix build caused by 40ca9ea80e
GitOrigin-RevId: b4868a08f6
|
5 years ago |
Megvii Engine Team
|
2c9fa7f650
|
feat(mge/profiler): add compatible and graphviz mode
GitOrigin-RevId: 40ca9ea80e
|
5 years ago |
Megvii Engine Team
|
3246ee5e87
|
perf(mge): use DepType::HOST_VALUE in trace when possible
GitOrigin-RevId: 5d47ed263f
|
5 years ago |
Megvii Engine Team
|
9389a805d8
|
fix(mge/oprmm): fix grad for collective comm
GitOrigin-RevId: e55dc47a6e
|
5 years ago |
Megvii Engine Team
|
6f581906a1
|
refactor(mge/profiler): refactor profiler
GitOrigin-RevId: 279aa779a6
|
5 years ago |
Megvii Engine Team
|
bd3b9cb6d4
|
fix(mge/oprmm): fix grad for collective comm
GitOrigin-RevId: 8e28f46c90
|
5 years ago |
Megvii Engine Team
|
b1ab3646f5
|
feat(imperative): add tensor sanity check
GitOrigin-RevId: 27243978e3
|
5 years ago |
Megvii Engine Team
|
939bf3b28d
|
fix(mge/imperative): impl hashable for nms and cond_take
GitOrigin-RevId: 56918db014
|
5 years ago |
Megvii Engine Team
|
3bbfef3009
|
fix(mge/imperative): impl hashable for coll-comm
GitOrigin-RevId: 76ab16a89b
|
5 years ago |
Megvii Engine Team
|
0537cb7471
|
chore(mge/imperative): fix BackwardGraph for jit.trace
GitOrigin-RevId: 16e86a21d7
|
5 years ago |
Megvii Engine Team
|
4101d5bca0
|
refactor(mge/imperative): add BackwardGraph.interpret
GitOrigin-RevId: bb3a59380e
|
5 years ago |
Megvii Engine Team
|
d4bad71190
|
feat(mge): add jit.trace
GitOrigin-RevId: ec647324c0
|
5 years ago |
Megvii Engine Team
|
236c7ec80b
|
refactor(imperative): remove unused opdef gen
GitOrigin-RevId: 509dbd0bb2
|
5 years ago |
Megvii Engine Team
|
c5e433b568
|
refactor(mge): cpp license
GitOrigin-RevId: 2f67f84724
|
5 years ago |
Megvii Engine Team
|
6b380e8965
|
feat(mge/imperative): run oss test and restore cmake list build items
GitOrigin-RevId: 11411b6964
|
5 years ago |