Megvii Engine Team
|
064c774f05
|
feat(imperative): impl hashable for SendRecv and add virtual input for Recv
GitOrigin-RevId: 5e8c27ac81
|
4 years ago |
Megvii Engine Team
|
798f7b3ef6
|
feat(imperative): add virtual_deps op
GitOrigin-RevId: 5845520989
|
4 years ago |
Megvii Engine Team
|
33e8879af4
|
feat(mge/quantization): support distributed qat
GitOrigin-RevId: c915c843b8
|
4 years ago |
Megvii Engine Team
|
8e11204ab6
|
perf(mge/module): optimize conv_bn qat module to improve performance
GitOrigin-RevId: 9415b83d9c
|
4 years ago |
Megvii Engine Team
|
51fa530d2a
|
fix(mge/interpreter): add check for invalid tensor ptr
GitOrigin-RevId: e8edcd92a4
|
4 years ago |
Megvii Engine Team
|
634de59061
|
feat(mge/imperative): add valid flag of `infer_output_attrs_fallible`
GitOrigin-RevId: b2b32774ee
|
4 years ago |
Megvii Engine Team
|
50c4daac85
|
feat(mge/interpreter): add async_level mechanism for Interpreter
GitOrigin-RevId: 8615a23b75
|
4 years ago |
Megvii Engine Team
|
82b0f67770
|
fix(mge/core): fix dtype promotion issue for quantized dtype
GitOrigin-RevId: 9d09c8fa6f
|
4 years ago |
Megvii Engine Team
|
8118a5946d
|
fix(mge/utils): fix get_oprs_seq of cgtools
GitOrigin-RevId: 366a56f4d5
|
4 years ago |
Megvii Engine Team
|
ae8c3c8163
|
feat(mge/functional): add python wrapper for fake quant opr
GitOrigin-RevId: 5f205198be
|
4 years ago |
Megvii Engine Team
|
2e53077915
|
fix(mge/trace): use xpux device when dump
GitOrigin-RevId: f37285f70e
|
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
|
094601e834
|
feat(mge/distributed): allow remote grad by using grad manager
GitOrigin-RevId: a890c206a5
|
4 years ago |
Megvii Engine Team
|
cae8c8a4a4
|
test(mge/parampack): add test for parampack
GitOrigin-RevId: f3e423dffc
|
4 years ago |
Megvii Engine Team
|
02438ee635
|
refactor(mge/distributed): use thread.Threading to create Server
GitOrigin-RevId: fc994411bf
|
4 years ago |
Megvii Engine Team
|
a1efd4d56d
|
fix(mge/parampack): fix param pack when no param left
GitOrigin-RevId: b88b876064
|
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
|
7cd846a570
|
fix(mge/batch_norm): fix batch_norm check when trace(symbolic=True)
GitOrigin-RevId: 2032eb5f7d
|
4 years ago |
Megvii Engine Team
|
120e719e17
|
refactor(imperative): refactor OpTrait methods and registration
remove unused op methods and enable OpTrait registered from seperated files, also fix cond_take's infer_output_attrs
GitOrigin-RevId: 134c8215ce
|
4 years ago |
Megvii Engine Team
|
c89209450d
|
ci(mge): remove blank line for windows doctest
GitOrigin-RevId: 09cf6518e8
|
4 years ago |
Megvii Engine Team
|
e082e27780
|
refactor(mge/data): Refactor megeninge.data.dataset
GitOrigin-RevId: 1d9c61ce70
|
4 years ago |
Megvii Engine Team
|
f04e0d777e
|
feat(mge/data): dpflow dataset, stream sampler and loader
GitOrigin-RevId: cbb4510a13
|
4 years ago |
Megvii Engine Team
|
be511a56f9
|
fix(mge/imperative): fix tensor astype failed for quantized type
GitOrigin-RevId: 383458acbf
|
4 years ago |
Megvii Engine Team
|
b309890c66
|
docs(mge): pytest for sphinx docstring
GitOrigin-RevId: 8bed12562a
|
4 years ago |
Megvii Engine Team
|
1b56851715
|
fix(mge/trace): fix graph option in trace
GitOrigin-RevId: 7bec84f56d
|
4 years ago |
Megvii Engine Team
|
1217801133
|
perf(mge): add opdef for broadcast
GitOrigin-RevId: 92f0af29eb
|
4 years ago |
Megvii Engine Team
|
fccb2510c4
|
perf(mge): add opdef for elemwise and batchnorm
GitOrigin-RevId: d51fad9867
|
4 years ago |
Megvii Engine Team
|
9ffde21236
|
feat(python): add dct python opr
GitOrigin-RevId: 3814dec267
|
4 years ago |
Megvii Engine Team
|
b327822994
|
feat(mge/grad_manager): add `clear_grad` method for GradManager
GitOrigin-RevId: aa9540e090
|
4 years ago |
Megvii Engine Team
|
2627e1f7df
|
fix(mge/grad_manager): allow multiple calls of `release`
GitOrigin-RevId: 38ca4c78ff
|
4 years ago |
Megvii Engine Team
|
a9f98e9c66
|
refactor(meg/internal): move interal codes back to megbrain
GitOrigin-RevId: b2dbda96be
|
4 years ago |
Megvii Engine Team
|
e8bf5bc002
|
fix(mge): ensure contiguous when passing tensor from graph rt to imperative rt
GitOrigin-RevId: e4d944343d
|
4 years ago |
Megvii Engine Team
|
3af1056377
|
feat(mge/grad): attach grad immediately
GitOrigin-RevId: e3a168c03a
|
4 years ago |
Megvii Engine Team
|
d8ac6c70a1
|
fix(quantize): fix quantize calibration dtype issue
GitOrigin-RevId: 667c99f469
|
4 years ago |
Megvii Engine Team
|
50d64bd333
|
fix(pytest/windows/impertive): open impertive pytest on windows
GitOrigin-RevId: 2e8f15607e
|
4 years ago |
Megvii Engine Team
|
dfc3935d06
|
fix(pytest/windows/impertive): fix windows wrong test skip
GitOrigin-RevId: ca0f04cfe7
|
4 years ago |
Megvii Engine Team
|
786399c5c9
|
fix(pytest/windows/impertive): fix impertive pytest failed on windows
GitOrigin-RevId: 02f4c0a0be
|
4 years ago |
Megvii Engine Team
|
215f88f373
|
fix(dnn/argmxx): fix argmxx on inf
GitOrigin-RevId: 740f67b73a
|
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
|
9748aebeea
|
refactor(mge): tensor_shape -> symbolic_shape
GitOrigin-RevId: 366dc048bf
|
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
|
5cc043f015
|
fix(imperative): sync command queue at exit
GitOrigin-RevId: c0d6aad275
|
4 years ago |
Megvii Engine Team
|
0075e6acb0
|
refactor(mge): refactor cross_entropy
GitOrigin-RevId: 1fac5b5b14
|
4 years ago |
Megvii Engine Team
|
f4b169322d
|
test(mgb/imperative): add adaptive pooling pytest
GitOrigin-RevId: c4dfed1f80
|
4 years ago |