Megvii Engine Team
|
abe3c165ba
|
feat(mge): tensorrt runtime opr
GitOrigin-RevId: 2fdd00adcb
|
5 years ago |
Megvii Engine Team
|
c53abcdf1d
|
chore(mge): minor improvements related to grad
GitOrigin-RevId: 102467d79d
|
5 years ago |
Megvii Engine Team
|
0a3ca25337
|
fix(mge): fix backward graph optimization
GitOrigin-RevId: 28fd00ac54
|
5 years ago |
Megvii Engine Team
|
c8e04ce4df
|
fix(imperative/interpreter): use info->h_value for host compute
Rather than use host_only Tensor and modify Put's behavior.
GitOrigin-RevId: f890d66acb
|
5 years ago |
Megvii Engine Team
|
a5a606792e
|
feat(imperative/interpreter): add more dispatch mode in apply_op
GitOrigin-RevId: 2663504470
|
5 years ago |
Megvii Engine Team
|
248d8bf0dc
|
feat(imperative/ops): improve infer attrs validate function
GitOrigin-RevId: 6fab3b1402
|
5 years ago |
Megvii Engine Team
|
8ed2077bcb
|
fix(imperative/interpreter): reset m_waitee on exception
GitOrigin-RevId: ecf5245d9c
|
5 years ago |
Megvii Engine Team
|
8fe865d88a
|
feat(imperative/ops): add infer_output_attrs for Reshape
GitOrigin-RevId: 9150d7f84d
|
5 years ago |
Megvii Engine Team
|
267d612723
|
fix(imperative/mini_graph): fix no_err value in shape infer
GitOrigin-RevId: 087d15db66
|
5 years ago |
Megvii Engine Team
|
bc3561c1ee
|
fix(imperative/ops): fix inplace_add's output desc, remove value
GitOrigin-RevId: b294eead0a
|
5 years ago |
Megvii Engine Team
|
55042195d4
|
chore(winograd): add Convolutionv2 param
GitOrigin-RevId: 1a9e2ea340
|
5 years ago |
Megvii Engine Team
|
e9a7be46b1
|
feat(mgb): adapt to imperative runtime
GitOrigin-RevId: 3bccc17b62
|
5 years ago |
Megvii Engine Team
|
998f71a8c9
|
perf(mge/imperative): do not busy wait in imperative's queue
GitOrigin-RevId: de8db51096
|
5 years ago |
Megvii Engine Team
|
278b2baa8c
|
perf(mge): add memory optimization for backward graph
precompute ops in forward to reduce saved tensor size
GitOrigin-RevId: d67043ba82
|
5 years ago |
Megvii Engine Team
|
a90c937d22
|
feat(interpreter): add command buffer for inplace
GitOrigin-RevId: 020d1e88d4
|
5 years ago |
Megvii Engine Team
|
6919127fd4
|
fix(mge): fix event pool destructor order issue and do full sync at python exit
GitOrigin-RevId: b9fa1f45b1
|
5 years ago |
Megvii Engine Team
|
c49427d15a
|
feat(imperative): add inplace add_update option in optimizer
GitOrigin-RevId: b8feb49321
|
5 years ago |
Megvii Engine Team
|
a85531dd0f
|
feat(mgb/opr): add tqt opr
GitOrigin-RevId: 49c62cd532
|
5 years ago |
Megvii Engine Team
|
aba0acc797
|
fix(sdk): add AssertEqual opr, fix dump_with_testcase_mge
GitOrigin-RevId: 6f797570b6
|
5 years ago |
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
|
d168cea4a7
|
feat(opr): add param(axis) for GetVarShape
feat(mge/imperative): GetVarShape support negative axis
GitOrigin-RevId: 30ce0758e6
|
5 years ago |
Megvii Engine Team
|
e9c036ccba
|
fix(mge/imperative): fix op specializations and tuple hash
GitOrigin-RevId: e3df93fd5b
|
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
|
9de1ea6a48
|
perf(imperative): add apply_on_physical_tensor for Elemwise
GitOrigin-RevId: 27087d90e4
|
5 years ago |
Megvii Engine Team
|
a78c11099b
|
fix(imperative): add param(axis) for GetVarShape
GitOrigin-RevId: 0b8f821929
|
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
|
bff0fc6172
|
fix(mge/interpreter): fix outputs check on async level0
GitOrigin-RevId: 46c6a2ce70
|
5 years ago |
Megvii Engine Team
|
93f4977c78
|
feat(mge/imperative): add thread name
GitOrigin-RevId: 61eece55b8
|
5 years ago |
Megvii Engine Team
|
7aa54b0ec6
|
feat(mge): enable memory swap and drop/recomputation
GitOrigin-RevId: c56c87c88b
|
5 years ago |
Megvii Engine Team
|
94dba16ff4
|
perf(mge/imperative): misc optimizations
GitOrigin-RevId: bbe7a10b00
|
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
|
51fa530d2a
|
fix(mge/interpreter): add check for invalid tensor ptr
GitOrigin-RevId: e8edcd92a4
|
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
|
5cc043f015
|
fix(imperative): sync command queue at exit
GitOrigin-RevId: c0d6aad275
|
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
|
c92044c5c4
|
refactor(mge/log): use info as default log level
GitOrigin-RevId: 3b7719a9fe
|
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
|
165798bf9c
|
fix(mge/build): move version.ld to src under project source root
GitOrigin-RevId: 04f39943c4
|
5 years ago |