Megvii Engine Team
|
5b697c714c
|
chore(imperative): remove unused reference to megbrain_graph
GitOrigin-RevId: 2c31f13bb8
|
4 years ago |
Megvii Engine Team
|
bb8f292816
|
refactor(mge/func): change `conv_execution_strategy` to `execution_strategy`
GitOrigin-RevId: 7aef9935ee
|
4 years ago |
Megvii Engine Team
|
30d6b4f65d
|
fix(mge): fix scalar parameter change to 1-dim parameter after backward and optimize
GitOrigin-RevId: 1794369a71
|
4 years ago |
Megvii Engine Team
|
05692d058f
|
refactor(functional): move matmul dot svd to math pkg
GitOrigin-RevId: 15eb08bacb
|
4 years ago |
Megvii Engine Team
|
cd02d7c85e
|
feat(mge/opr): support deformable conv2d/psroi_pooling
GitOrigin-RevId: 501cadda76
|
4 years ago |
Megvii Engine Team
|
e82fa4ec23
|
fix(gopt): using new_inp for build_chain in DelayBroadcast pass
GitOrigin-RevId: efc6377197
|
4 years ago |
Megvii Engine Team
|
4485e780ae
|
fix(mge/functional): fix linspace device and open other trace tests
GitOrigin-RevId: 4667c4adec
|
4 years ago |
Megvii Engine Team
|
938152af63
|
fix(mge/functional): convert input type to float32 for more elemwise op
GitOrigin-RevId: cf3bf8cb80
|
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
|
e3a3e0cd3c
|
feat(mge/functional): add warp_affine
GitOrigin-RevId: 2b333ccd12
|
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
|
df976782fa
|
feat(mge/functional): add matinv
GitOrigin-RevId: d4fa8a8277
|
4 years ago |
Megvii Engine Team
|
ca1d74ad3f
|
fix(mge/utils): fix types import error
GitOrigin-RevId: d2d4bd2725
|
4 years ago |
Megvii Engine Team
|
abe3c165ba
|
feat(mge): tensorrt runtime opr
GitOrigin-RevId: 2fdd00adcb
|
4 years ago |
Megvii Engine Team
|
e9a7be46b1
|
feat(mgb): adapt to imperative runtime
GitOrigin-RevId: 3bccc17b62
|
4 years ago |
Megvii Engine Team
|
5c7d48cdb9
|
fix(mge/functional): fix tensor split
GitOrigin-RevId: 0a112ab0bd
|
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
|
44742e32e9
|
fix(mge/api): check input dim of dot and mark output as scalar
GitOrigin-RevId: a3ba7e099c
|
4 years ago |
Megvii Engine Team
|
dd9f54cdfa
|
refactor(mge): migrate to new core implementation
* remove dispatcher/interpreter python wrapper
* rename tensor_wrapper to array_method
GitOrigin-RevId: b8a402c2be
|
4 years ago |
Megvii Engine Team
|
60c7d62a4c
|
refactor(imperative): remove multidispatch, raw_tensor, register
GitOrigin-RevId: ca5a6ed8eb
|
4 years ago |
Megvii Engine Team
|
87f4b46e3f
|
perf(mge/imperative): move convert_inputs from python to C++
GitOrigin-RevId: baef3d348c
|
4 years ago |
Megvii Engine Team
|
c294b9d18b
|
refactor(mge/tensor): remove old implementation
remove core.tensor, raw_tensor,TensorWrapper
avoid create tensor with zero-stride numpy ndarray
GitOrigin-RevId: 4fe5c4c5ba
|
4 years ago |
Megvii Engine Team
|
15e8e7df76
|
refactor(mge/imperative): migrate to new core implementation
* swap/drop
* config/set_async_level
* _dev_tensor
* sync
GitOrigin-RevId: 850fb98852
|
4 years ago |
Megvii Engine Team
|
147cef52a0
|
refactor(mge/imperative): implement new tensor system
GitOrigin-RevId: 2dd4e460ac
|
4 years ago |
Megvii Engine Team
|
69e3e32240
|
feat(imperative): auto generated opdef header and python binding
GitOrigin-RevId: d2f22ad5fe
|
4 years ago |
Megvii Engine Team
|
638ab52fdc
|
feat(mge/imperative): simulates scalar
GitOrigin-RevId: e81630e256
|
4 years ago |
Megvii Engine Team
|
94dba16ff4
|
perf(mge/imperative): misc optimizations
GitOrigin-RevId: bbe7a10b00
|
4 years ago |
Megvii Engine Team
|
442058aead
|
feat(mge/quantize): add batch matmul activation module for inference
GitOrigin-RevId: f6dbd1f4c0
|
4 years ago |
Megvii Engine Team
|
9cb3c07cd4
|
feat(mge/functional): add elemwise mode support string input
GitOrigin-RevId: 57be5cec7b
|
4 years ago |
Megvii Engine Team
|
8fad00a158
|
feat(mge/functional/nn): add conv1d padding
GitOrigin-RevId: 1bbfd36b96
|
4 years ago |
Megvii Engine Team
|
e0e1896450
|
docs(mge): formatting math in docstring
GitOrigin-RevId: e6447a2786
|
4 years ago |
Megvii Engine Team
|
28dbadf729
|
feat(imperative): add remap opr
GitOrigin-RevId: 1fa7e18d4b
|
4 years ago |
Megvii Engine Team
|
1e0fb127d1
|
perf(mge/functional): reduce matmul python overhead
GitOrigin-RevId: 738d0da10e
|
4 years ago |
Megvii Engine Team
|
c89209450d
|
ci(mge): remove blank line for windows doctest
GitOrigin-RevId: 09cf6518e8
|
4 years ago |
Megvii Engine Team
|
b309890c66
|
docs(mge): pytest for sphinx docstring
GitOrigin-RevId: 8bed12562a
|
4 years ago |
Megvii Engine Team
|
fccb2510c4
|
perf(mge): add opdef for elemwise and batchnorm
GitOrigin-RevId: d51fad9867
|
4 years ago |
Megvii Engine Team
|
1fa143ce87
|
refactor(mge/functional): matmul supports symbolic shape, batched mv multiply
GitOrigin-RevId: c4d8cf3306
|
4 years ago |
Megvii Engine Team
|
95f6b53183
|
feat(opr/nvof): add nvof operator
This reverts commit 18b84072ac .
GitOrigin-RevId: 3b7622784d
|
4 years ago |
Megvii Engine Team
|
8acc3acf70
|
fix(mge/functional): do not del anything
GitOrigin-RevId: 006c3a34ac
|
4 years ago |
Megvii Engine Team
|
5ee1a1c45c
|
fix(mge/functional): del graph in __init__
GitOrigin-RevId: 4a62f39546
|
4 years ago |
Megvii Engine Team
|
6701346398
|
docs(mge): fix some docstring format problem
GitOrigin-RevId: cbc5ab04b3
|
4 years ago |
Megvii Engine Team
|
0075e6acb0
|
refactor(mge): refactor cross_entropy
GitOrigin-RevId: 1fac5b5b14
|
4 years ago |
Megvii Engine Team
|
37e56f4b04
|
fix(mge/functional): fix clip under trace(symbolic=True)
GitOrigin-RevId: 5b6f537327
|
4 years ago |
Megvii Engine Team
|
af349d6102
|
fix(mge/functional): fix op mismatch when tracing NMSKeep
GitOrigin-RevId: e8f2cbb755
|
4 years ago |
Megvii Engine Team
|
d502e79f85
|
feat(mge): make F.norm numpy compatible
GitOrigin-RevId: 4e5236171c
|
4 years ago |
Megvii Engine Team
|
d31a4fff73
|
refactor(mge): use logsumexp in cross_entropy
GitOrigin-RevId: 4a14aabb94
|
4 years ago |
Megvii Engine Team
|
20e93630f2
|
feat(mge): rename batch_norm2d -> batch_norm
GitOrigin-RevId: 253e8564ea
|
4 years ago |
Megvii Engine Team
|
aa62672629
|
feat(mge): make F.eye numpy compatible
GitOrigin-RevId: fee32537b4
|
4 years ago |
Megvii Engine Team
|
7589bce454
|
feat(mge): do not export F.nn.sync_batch_norm
GitOrigin-RevId: aa05e3aac4
|
4 years ago |