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 |