Megvii Engine Team
|
d970b85deb
|
feat(functional): add conv3d
GitOrigin-RevId: c06ac77c6c
|
4 years ago |
Megvii Engine Team
|
46cad4d3c8
|
feat(functional/ops): add _assert_equal
GitOrigin-RevId: b7ce4158b7
|
4 years ago |
Megvii Engine Team
|
dcfb6a537e
|
refactor(mge/functional): move functional api
GitOrigin-RevId: 9cd3e09996
|
4 years ago |
Megvii Engine Team
|
0035efa636
|
fix(mge): limit task queue size
GitOrigin-RevId: 9481d38914
|
4 years ago |
Megvii Engine Team
|
ad87f78a14
|
chore(imperative): refine tblgen for generating op name
GitOrigin-RevId: f47ceae726
|
4 years ago |
Megvii Engine Team
|
0a191f8d48
|
fix(misc): some fixes to make faster_rcnn run in no-symbolic-trace
GitOrigin-RevId: 0fc40ad9c8
|
4 years ago |
Megvii Engine Team
|
f95bb7b7c9
|
perf(mge/profiler): reduce profiler overhead
GitOrigin-RevId: 4e1fff9dc4
|
4 years ago |
Megvii Engine Team
|
6fb19b664b
|
feat(imperative/src): name operators automatically when tracing
GitOrigin-RevId: ff8eb003c5
|
4 years ago |
Megvii Engine Team
|
cbff4d7c1a
|
fix(imperative/ops): fix infer_output_attrs_fallible for reshape
GitOrigin-RevId: a93567d79a
|
4 years ago |
Megvii Engine Team
|
cd02d7c85e
|
feat(mge/opr): support deformable conv2d/psroi_pooling
GitOrigin-RevId: 501cadda76
|
4 years ago |
Megvii Engine Team
|
fc8d13cda3
|
fix(build): fix linkage on ubuntu1604
GitOrigin-RevId: ee2a8360b0
|
4 years ago |
Megvii Engine Team
|
194660462c
|
feat(mge/funcitonal): add cvt_color opr python interface
GitOrigin-RevId: 29e069fb23
|
4 years ago |
Megvii Engine Team
|
dbb3dd681f
|
refactor(profiler): integrate profiler into interpreter
GitOrigin-RevId: ccc984acbd
|
4 years ago |
Megvii Engine Team
|
e3a3e0cd3c
|
feat(mge/functional): add warp_affine
GitOrigin-RevId: 2b333ccd12
|
4 years ago |
Megvii Engine Team
|
fe1680b378
|
fix(imperative/ops): improve infer_output_attrs for broadcast
GitOrigin-RevId: 6b7ed55769
|
4 years ago |
Megvii Engine Team
|
c3ba02801a
|
chore(mge): correct license headers
GitOrigin-RevId: 2dd4438f90
|
4 years ago |
Megvii Engine Team
|
364afec033
|
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
|
4 years ago |
Megvii Engine Team
|
4aaae9950b
|
feat(mge/functional): add python wrapper to resize opr
GitOrigin-RevId: b7cc6dd829
|
4 years ago |
Megvii Engine Team
|
d04b4bc006
|
fix(interp): thread safety for drop and swapout
GitOrigin-RevId: 7684f160bf
|
4 years ago |
Megvii Engine Team
|
3eb529d692
|
refactor(interpreter): recognize recomp on main thread rather than worker
GitOrigin-RevId: 4ba3942ce4
|
4 years ago |
Megvii Engine Team
|
df976782fa
|
feat(mge/functional): add matinv
GitOrigin-RevId: d4fa8a8277
|
4 years ago |
Megvii Engine Team
|
abe3c165ba
|
feat(mge): tensorrt runtime opr
GitOrigin-RevId: 2fdd00adcb
|
4 years ago |
Megvii Engine Team
|
c53abcdf1d
|
chore(mge): minor improvements related to grad
GitOrigin-RevId: 102467d79d
|
4 years ago |
Megvii Engine Team
|
0a3ca25337
|
fix(mge): fix backward graph optimization
GitOrigin-RevId: 28fd00ac54
|
4 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
|
4 years ago |
Megvii Engine Team
|
a5a606792e
|
feat(imperative/interpreter): add more dispatch mode in apply_op
GitOrigin-RevId: 2663504470
|
4 years ago |
Megvii Engine Team
|
248d8bf0dc
|
feat(imperative/ops): improve infer attrs validate function
GitOrigin-RevId: 6fab3b1402
|
4 years ago |
Megvii Engine Team
|
8ed2077bcb
|
fix(imperative/interpreter): reset m_waitee on exception
GitOrigin-RevId: ecf5245d9c
|
4 years ago |
Megvii Engine Team
|
8fe865d88a
|
feat(imperative/ops): add infer_output_attrs for Reshape
GitOrigin-RevId: 9150d7f84d
|
4 years ago |
Megvii Engine Team
|
267d612723
|
fix(imperative/mini_graph): fix no_err value in shape infer
GitOrigin-RevId: 087d15db66
|
4 years ago |
Megvii Engine Team
|
bc3561c1ee
|
fix(imperative/ops): fix inplace_add's output desc, remove value
GitOrigin-RevId: b294eead0a
|
4 years ago |
Megvii Engine Team
|
55042195d4
|
chore(winograd): add Convolutionv2 param
GitOrigin-RevId: 1a9e2ea340
|
4 years ago |
Megvii Engine Team
|
e9a7be46b1
|
feat(mgb): adapt to imperative runtime
GitOrigin-RevId: 3bccc17b62
|
4 years ago |
Megvii Engine Team
|
998f71a8c9
|
perf(mge/imperative): do not busy wait in imperative's queue
GitOrigin-RevId: de8db51096
|
4 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
|
4 years ago |
Megvii Engine Team
|
a90c937d22
|
feat(interpreter): add command buffer for inplace
GitOrigin-RevId: 020d1e88d4
|
4 years ago |
Megvii Engine Team
|
6919127fd4
|
fix(mge): fix event pool destructor order issue and do full sync at python exit
GitOrigin-RevId: b9fa1f45b1
|
4 years ago |
Megvii Engine Team
|
c49427d15a
|
feat(imperative): add inplace add_update option in optimizer
GitOrigin-RevId: b8feb49321
|
4 years ago |
Megvii Engine Team
|
a85531dd0f
|
feat(mgb/opr): add tqt opr
GitOrigin-RevId: 49c62cd532
|
4 years ago |
Megvii Engine Team
|
aba0acc797
|
fix(sdk): add AssertEqual opr, fix dump_with_testcase_mge
GitOrigin-RevId: 6f797570b6
|
4 years ago |
Megvii Engine Team
|
dc2507453b
|
feat(mge): add python custom op
GitOrigin-RevId: 35da0bb301
|
4 years ago |
Megvii Engine Team
|
14d8b709e1
|
perf(mge/imperative): add mini graph to partially replace proxy graph
GitOrigin-RevId: 73e2529ba5
|
4 years ago |
Megvii Engine Team
|
147cef52a0
|
refactor(mge/imperative): implement new tensor system
GitOrigin-RevId: 2dd4e460ac
|
4 years ago |
Megvii Engine Team
|
d168cea4a7
|
feat(opr): add param(axis) for GetVarShape
feat(mge/imperative): GetVarShape support negative axis
GitOrigin-RevId: 30ce0758e6
|
4 years ago |
Megvii Engine Team
|
e9c036ccba
|
fix(mge/imperative): fix op specializations and tuple hash
GitOrigin-RevId: e3df93fd5b
|
4 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
|
4 years ago |
Megvii Engine Team
|
9a79342d35
|
refactor(mge/virtualdep): remove virtualdep opdef
GitOrigin-RevId: bbe3ae3fa3
|
4 years ago |
Megvii Engine Team
|
88b402ef62
|
feat(mge/trace): tracing use id to set priority
GitOrigin-RevId: 6e1f1ece0e
|
4 years ago |
Megvii Engine Team
|
9de1ea6a48
|
perf(imperative): add apply_on_physical_tensor for Elemwise
GitOrigin-RevId: 27087d90e4
|
4 years ago |
Megvii Engine Team
|
a78c11099b
|
fix(imperative): add param(axis) for GetVarShape
GitOrigin-RevId: 0b8f821929
|
4 years ago |