Megvii Engine Team
|
86b69cacd0
|
fix(dnn): fixes for int4
GitOrigin-RevId: 845e164fd3
|
4 years ago |
Megvii Engine Team
|
d2673c5abf
|
fix(ci/windows): add windows cuda test
GitOrigin-RevId: 706be83032
|
4 years ago |
Megvii Engine Team
|
2d6827c168
|
fix(mgb/windows): temporary workround on cuda-windows python exit
code(127), as windows cuda driver unloading before atexit function
may remove this after upgrade cuda runtime
GitOrigin-RevId: cac37ca3dd
|
4 years ago |
Megvii Engine Team
|
241b35a697
|
refactor(ops): remove BackwardGraph op
GitOrigin-RevId: eda20e5760
|
4 years ago |
Megvii Engine Team
|
1525a02530
|
feat(mge/module): add python wrapper for unfold
GitOrigin-RevId: 562103186f
|
4 years ago |
Megvii Engine Team
|
4f240ec2d3
|
refactor(mge/jit): make trace return any kind of output
GitOrigin-RevId: fd1265c661
|
4 years ago |
Megvii Engine Team
|
13b15fb08c
|
feat(megbrain): add correlation opr
GitOrigin-RevId: 6d44598891
|
4 years ago |
Megvii Engine Team
|
601a33a84d
|
refactor(mge/dtr): update dtr api
GitOrigin-RevId: dc366c65be
|
4 years ago |
Megvii Engine Team
|
c269a690e5
|
fix(mge/dtr): filter bad ops
GitOrigin-RevId: 380262dca0
|
4 years ago |
Megvii Engine Team
|
9b69a02fb2
|
fix(imperative/tensor): init m_offset when constructing a Tensor with DeviceTensorND
GitOrigin-RevId: b340e27c47
|
4 years ago |
Megvii Engine Team
|
346d242051
|
fix(param_pack): impl param pack concat in imperative_rt
GitOrigin-RevId: 91edd9c0bf
|
4 years ago |
Megvii Engine Team
|
9c17cfc45d
|
fix(imperative/ops): add check_dtype for Elemwise in infer_attrs
GitOrigin-RevId: c777855753
|
4 years ago |
Megvii Engine Team
|
7af1ec66db
|
fix(tensor): del valid tensors when compnode finalizing
GitOrigin-RevId: bace1f2b51
|
4 years ago |
Megvii Engine Team
|
516769391c
|
fix(mge/imperative): fix proxy graph comp_node finalize
GitOrigin-RevId: 22ad0d0fc6
|
5 years ago |
Megvii Engine Team
|
13e6ea349d
|
feat(imperative/opr): rebase rng refactoring to dev & add python module
GitOrigin-RevId: ee5984c52d
|
4 years ago |
Megvii Engine Team
|
cded8ef1a6
|
feat(imperative): add rng opdef
GitOrigin-RevId: b62dcea7f5
|
5 years ago |
Megvii Engine Team
|
e18afa0b05
|
feat(mge/module): python wrapper for conv_transpose3d
GitOrigin-RevId: 61097b8713
|
4 years ago |
Megvii Engine Team
|
1a7112997c
|
feat(opr-mm): add backend argument for remote send/recv
GitOrigin-RevId: 841a0e45ab
|
4 years ago |
Megvii Engine Team
|
fe99cdc794
|
feat(interpreter): add dynamic sublinear
GitOrigin-RevId: 7de54fe743
|
4 years ago |
Megvii Engine Team
|
a3ab9f1d41
|
feat(blob_manager): make all memory allocation go through blob manager
GitOrigin-RevId: f79155e5c3
|
4 years ago |
Megvii Engine Team
|
58ebb26156
|
fix(imperative/tensor): fix ConstTensorCache
GitOrigin-RevId: 0767bcfa28
|
4 years ago |
Megvii Engine Team
|
4f0e6eae35
|
fix(mge/trace): re-open gopt level in trace
GitOrigin-RevId: 5ebc712690
|
4 years ago |
Megvii Engine Team
|
fa1ca0ea6c
|
fix(imperative/opr): fix apply_on_var_node for broadcast
GitOrigin-RevId: 686fff4f73
|
4 years ago |
Megvii Engine Team
|
221ec38a48
|
feat(imperative): reduce profiler overhead
GitOrigin-RevId: dded9d9391
|
4 years ago |
Megvii Engine Team
|
c8697a7005
|
feat(imperative/src): python wrapper for cambricon and atlas runtime opr
GitOrigin-RevId: bd969d1339
|
4 years ago |
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
|
5 years ago |
Megvii Engine Team
|
f95bb7b7c9
|
perf(mge/profiler): reduce profiler overhead
GitOrigin-RevId: 4e1fff9dc4
|
5 years ago |
Megvii Engine Team
|
6fb19b664b
|
feat(imperative/src): name operators automatically when tracing
GitOrigin-RevId: ff8eb003c5
|
5 years ago |
Megvii Engine Team
|
cbff4d7c1a
|
fix(imperative/ops): fix infer_output_attrs_fallible for reshape
GitOrigin-RevId: a93567d79a
|
5 years ago |
Megvii Engine Team
|
cd02d7c85e
|
feat(mge/opr): support deformable conv2d/psroi_pooling
GitOrigin-RevId: 501cadda76
|
5 years ago |
Megvii Engine Team
|
fc8d13cda3
|
fix(build): fix linkage on ubuntu1604
GitOrigin-RevId: ee2a8360b0
|
5 years ago |
Megvii Engine Team
|
194660462c
|
feat(mge/funcitonal): add cvt_color opr python interface
GitOrigin-RevId: 29e069fb23
|
5 years ago |
Megvii Engine Team
|
dbb3dd681f
|
refactor(profiler): integrate profiler into interpreter
GitOrigin-RevId: ccc984acbd
|
5 years ago |
Megvii Engine Team
|
e3a3e0cd3c
|
feat(mge/functional): add warp_affine
GitOrigin-RevId: 2b333ccd12
|
5 years ago |
Megvii Engine Team
|
fe1680b378
|
fix(imperative/ops): improve infer_output_attrs for broadcast
GitOrigin-RevId: 6b7ed55769
|
5 years ago |
Megvii Engine Team
|
c3ba02801a
|
chore(mge): correct license headers
GitOrigin-RevId: 2dd4438f90
|
5 years ago |
Megvii Engine Team
|
364afec033
|
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
|
5 years ago |
Megvii Engine Team
|
4aaae9950b
|
feat(mge/functional): add python wrapper to resize opr
GitOrigin-RevId: b7cc6dd829
|
5 years ago |
Megvii Engine Team
|
d04b4bc006
|
fix(interp): thread safety for drop and swapout
GitOrigin-RevId: 7684f160bf
|
5 years ago |
Megvii Engine Team
|
3eb529d692
|
refactor(interpreter): recognize recomp on main thread rather than worker
GitOrigin-RevId: 4ba3942ce4
|
5 years ago |
Megvii Engine Team
|
df976782fa
|
feat(mge/functional): add matinv
GitOrigin-RevId: d4fa8a8277
|
5 years ago |
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 |