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
|
a09fc5f784
|
fix(mgb/serialization): disable inplace arith graph opt in graph load
GitOrigin-RevId: d63baf8356
|
4 years ago |
Megvii Engine Team
|
43b4d4a430
|
feat(imperative/test): add more correctness tests
GitOrigin-RevId: 2ef6027bb1
|
4 years ago |
Megvii Engine Team
|
91a3580f75
|
refactor(mge/cgtools): remove load_and_inference and use GraphInference
GitOrigin-RevId: 0e688ebd59
|
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
|
334eda8717
|
refactor(mge): test trace inside opr_test
GitOrigin-RevId: 2cf1135c1c
|
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
|
412d1f0cdc
|
fix(mge/tensor): implement abtract method to fix lint errors
GitOrigin-RevId: d53f2eac6a
|
4 years ago |
Megvii Engine Team
|
dbb3dd681f
|
refactor(profiler): integrate profiler into interpreter
GitOrigin-RevId: ccc984acbd
|
4 years ago |
Megvii Engine Team
|
ff05667b48
|
test(dist): refactor distributed test with fixtures
GitOrigin-RevId: e69acb72da
|
4 years ago |
Megvii Engine Team
|
9fb8444d24
|
fix(imperative): catch python exception in c++
GitOrigin-RevId: 16a2abfdad
|
4 years ago |
Megvii Engine Team
|
e3a3e0cd3c
|
feat(mge/functional): add warp_affine
GitOrigin-RevId: 2b333ccd12
|
4 years ago |
Megvii Engine Team
|
8e50a6daa7
|
feat(mge/utils): add GraphInference in cgtools
GitOrigin-RevId: 72f2201169
|
4 years ago |
Megvii Engine Team
|
adc49de803
|
feat(mge/imperative): expose c++ tensor reference count
GitOrigin-RevId: 1940881adc
|
4 years ago |
Megvii Engine Team
|
27a9326c66
|
fix(mge/module): fix frozen batch norm
GitOrigin-RevId: 143d468a37
|
4 years ago |
Megvii Engine Team
|
364afec033
|
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
|
4 years ago |
Megvii Engine Team
|
9a09f761bb
|
fix(whl): append buildid to wheel version
GitOrigin-RevId: 00dabfa147
|
4 years ago |
Megvii Engine Team
|
af5ab4d976
|
fix(imperative/tensor): fix Py_DECREF on uninitialized pointer
GitOrigin-RevId: 889255537e
|
4 years ago |
Megvii Engine Team
|
10d9d77e41
|
refactor(mge/distributed): remove get_free_ports func
GitOrigin-RevId: b85c4885be
|
4 years ago |
Megvii Engine Team
|
0adf49b137
|
fix(mge/distributed): fix deadlock by mixing thread and fork
GitOrigin-RevId: c138cb9c28
|
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
|
2eed7d83be
|
docs(mge): restore autodiff.Function docs
GitOrigin-RevId: cbd84df916
|
4 years ago |
Megvii Engine Team
|
5100317660
|
feat(mge): make Function available at autodiff.Function
GitOrigin-RevId: 7c9ab65351
|
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
|
c418d3cd95
|
fix(mge/data): add timeout event
GitOrigin-RevId: 43f2ba1456
|
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
|
ea8eb4cf72
|
feat(mge/distributed): scalar support for distributed functions
GitOrigin-RevId: 53f3575baf
|
4 years ago |
Megvii Engine Team
|
b83c77e105
|
test(mge/test_save_load): remove test model file
GitOrigin-RevId: 893b82b1c9
|
4 years ago |
Megvii Engine Team
|
c522991385
|
test(mge/core): add host compute test, fix async_level and swap/drop test
GitOrigin-RevId: f9b5a83507
|
4 years ago |
Megvii Engine Team
|
6295f843a5
|
fix(mge/tensor): fix mem leak when getting value in trace
GitOrigin-RevId: 8bca2e7380
|
4 years ago |
Megvii Engine Team
|
7191c4bd9f
|
fix(mge/data): support timeout for serial stream dataloader
GitOrigin-RevId: 1ae5a8cfda
|
4 years ago |
Megvii Engine Team
|
b7ed0cb850
|
fix(mge/quantization): replace `_reset` with "=" in observer
GitOrigin-RevId: ed6af9b98d
|
4 years ago |
Megvii Engine Team
|
5d0c9bef4c
|
chore(mge/version): change dev version to 1.3
GitOrigin-RevId: b5e5327a46
|
4 years ago |
Megvii Engine Team
|
0b4918b25e
|
test(mge/quantization): classmethod `from_float_module` of qat module
GitOrigin-RevId: 95c3d45f83
|
4 years ago |
Megvii Engine Team
|
e9a7be46b1
|
feat(mgb): adapt to imperative runtime
GitOrigin-RevId: 3bccc17b62
|
4 years ago |
Megvii Engine Team
|
08cc10324e
|
fix(imperative): fix persistent_cache
GitOrigin-RevId: 8f7bb5899f
|
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
|
5c7d48cdb9
|
fix(mge/functional): fix tensor split
GitOrigin-RevId: 0a112ab0bd
|
4 years ago |
Megvii Engine Team
|
82847bc70a
|
docs(mge): add conv2d output shape computation formula
GitOrigin-RevId: 7ee476a80d
|
4 years ago |
Megvii Engine Team
|
d3bfb0e983
|
fix(mge): fix trace exit code and reformat
GitOrigin-RevId: 145c06b7e7
|
4 years ago |
Megvii Engine Team
|
23b9a98f5e
|
fix(mge): fix sublnear cuda and mem leak
GitOrigin-RevId: 82091ec9a6
|
4 years ago |
Megvii Engine Team
|
c70a49ed2c
|
fix(mge): correct trace outputs when grad does copy
GitOrigin-RevId: 65c8956a7d
|
4 years ago |
Megvii Engine Team
|
d4ada69d3b
|
refactor(mge): trace exception in compiled info
GitOrigin-RevId: 508f5463b9
|
4 years ago |
Megvii Engine Team
|
c9c3429a6b
|
refactor(mge): fix sublinear
GitOrigin-RevId: 5bb0383781
|
4 years ago |