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
13e6ea349d
feat(imperative/opr): rebase rng refactoring to dev & add python module
GitOrigin-RevId: ee5984c52d
4 years ago
Megvii Engine Team
40bab1ed66
feat(log): opt log, enable mgb sdk log at opt build
more info: 16cd674c56
* change MGE_OVERRIDE_LOG_LEVEL to RUNTIME_OVERRIDE_LOG_LEVEL
* use ::std::getenv not MGB_GETENV for special ENV
GitOrigin-RevId: ee0f9c0f72e627c331c00100f6a21adc927081df
4 years ago
Megvii Engine Team
e18afa0b05
feat(mge/module): python wrapper for conv_transpose3d
GitOrigin-RevId: 61097b8713
4 years ago
Megvii Engine Team
f16c9eb9d7
refactor(mgb): simplify fast run proccess with storing algo desc instead of algo name
GitOrigin-RevId: 04b86fde50
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
8585aa61ad
fix(mgb): fix fast run crash when profile heuristic strategy
GitOrigin-RevId: 6046a2db0c
4 years ago
Megvii Engine Team
62c394ca91
feat(cuda/comp_node): enable to query adjacent free blocks size
GitOrigin-RevId: 0cd1d202dd
4 years ago
Megvii Engine Team
fd61f09540
feat(cuda/comp_node): enable to directly query memory status
GitOrigin-RevId: 23c1af9a55
4 years ago
Megvii Engine Team
1310ad4926
feat(dnn): add memory manager for enflame
GitOrigin-RevId: 72a810d959
5 years ago
Megvii Engine Team
2fd3fa8834
feat(cmake): update for enflame cmake compile
GitOrigin-RevId: 3c3c6b3462
5 years ago
Megvii Engine Team
9f2af2099c
feat(mgb): add enflame comp node
GitOrigin-RevId: 478c8538aa
5 years ago
Megvii Engine Team
ef18a640bc
fix(mgb/core): fix allocate_task_block_unsafe in thread_impl_1.h
GitOrigin-RevId: 21b92fafc5
4 years ago
Megvii Engine Team
55974e8cf9
feat(log): opt log
* opt log at release mode
* add MGE_OVERRIDE_LOG_LEVEL for runtime debug
//! env to config LogLevel
//! DEBUG = 0, INFO = 1, WARN = 2, ERROR = 3, NO_LOG = 4
//! for example , export MGE_OVERRIDE_LOG_LEVEL=0, means set LogLevel to DEBUG
GitOrigin-RevId: 16cd674c56
4 years ago
Megvii Engine Team
e19b9af19b
feat(imperative): add bit combined enum to python C extension
GitOrigin-RevId: 92307dd2ca
4 years ago
Megvii Engine Team
a3ea1f153c
feat(mgb/opr): add fast profile and combined Execution strategy
GitOrigin-RevId: 843dc3a790
5 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
0035efa636
fix(mge): limit task queue size
GitOrigin-RevId: 9481d38914
4 years ago
Megvii Engine Team
b9bbf802bd
fix(mgb/opr): fix string append error
GitOrigin-RevId: 295a941edd
4 years ago
Megvii Engine Team
ad87f78a14
chore(imperative): refine tblgen for generating op name
GitOrigin-RevId: f47ceae726
4 years ago
Megvii Engine Team
6fb19b664b
feat(imperative/src): name operators automatically when tracing
GitOrigin-RevId: ff8eb003c5
5 years ago
Megvii Engine Team
b6376c83a0
fix(mgb/comp_node): use extra physical device to decide whether reuse existed CompNodeImpl
GitOrigin-RevId: 5dddc68a84
5 years ago
Megvii Engine Team
cd02d7c85e
feat(mge/opr): support deformable conv2d/psroi_pooling
GitOrigin-RevId: 501cadda76
5 years ago
Megvii Engine Team
f3378100f9
feat(mgb): enable output dynamic memory alloc
GitOrigin-RevId: c809629034
5 years ago
Megvii Engine Team
a09fc5f784
fix(mgb/serialization): disable inplace arith graph opt in graph load
GitOrigin-RevId: d63baf8356
5 years ago
Megvii Engine Team
c51a687cef
chore(mge): update copyright years
GitOrigin-RevId: 46104ac891
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
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
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
45e206020a
feat(mgb/tensor): add proxy_to_comp_node for HostTensorND
GitOrigin-RevId: 8f03dbcbd1
5 years ago
Megvii Engine Team
e9a7be46b1
feat(mgb): adapt to imperative runtime
GitOrigin-RevId: 3bccc17b62
5 years ago
Megvii Engine Team
62637fc46e
refactor(mgb/utils): allow user set different spin time for ASyncSCQueue
GitOrigin-RevId: 1c6c4a7a16
5 years ago
Megvii Engine Team
c3c1e46df7
chore(mgblar): delete winograd transform switch
GitOrigin-RevId: aa4051541d
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
7f3f9a94ae
feat(mgb/core): add shape hint for graph optimization
GitOrigin-RevId: eaad25a7ef
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
14d8b709e1
perf(mge/imperative): add mini graph to partially replace proxy graph
GitOrigin-RevId: 73e2529ba5
5 years ago
Megvii Engine Team
15dd5e1a0f
fix(mgb/core): fix memory management release cambricon var issue
GitOrigin-RevId: abf881978c
5 years ago
Megvii Engine Team
469d0808f5
fix(asan/thread): fix asan issue
GitOrigin-RevId: 8e2b093fb1
5 years ago
Megvii Engine Team
a78c11099b
fix(imperative): add param(axis) for GetVarShape
GitOrigin-RevId: 0b8f821929
5 years ago
Megvii Engine Team
368c18607f
fix(mgb/jit): find cuda include path correctly
GitOrigin-RevId: 5ced4206a7
5 years ago
Megvii Engine Team
cf53d9e0f8
fix(mgb/tensor): do tensor overlap check only when d2d and h2h
GitOrigin-RevId: 9125936a35
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
3bf73ff16f
feat(dnn): add cuda preprocess fusion
GitOrigin-RevId: d789c99e59
5 years ago