Megvii Engine Team
|
3eb0505f9b
|
feat(imperative): add support for quantized conv transpose2d
GitOrigin-RevId: ffd6431299
|
3 years ago |
Megvii Engine Team
|
a83098890b
|
test(mge/utils): cover all test data
GitOrigin-RevId: e676476b9d
|
3 years ago |
Megvii Engine Team
|
0b4a767965
|
feat(mge/distributed): enable uint8 for collective communication
GitOrigin-RevId: 3305c0cf14
|
3 years ago |
Megvii Engine Team
|
3f3a256e0f
|
fix(mge/functional): fix conv* dtype promotion
GitOrigin-RevId: 3f03790cfc
|
3 years ago |
Megvii Engine Team
|
f76a2cc2c6
|
feat(mge/opr): add silu and gelu
GitOrigin-RevId: 75aa42947e
|
3 years ago |
Megvii Engine Team
|
f141159088
|
refactor(mge): loose the error bound of fastrun
GitOrigin-RevId: 9bf9b9d4ca
|
3 years ago |
Megvii Engine Team
|
1f0436967c
|
refactor(mge/distributed): using nccl as default in distributed training
GitOrigin-RevId: 81268e84bc
|
3 years ago |
Megvii Engine Team
|
207a346351
|
chore(mge): run get_device_count("gpu") in subprocess
GitOrigin-RevId: 0f0dc001cf
|
4 years ago |
Xinda Huang
|
c90fa087ea
|
test(mge): delete test_external.py
|
3 years ago |
Megvii Engine Team
|
b2944559a8
|
fix(imperative/module): remove ``__getattribute__`` method in module
GitOrigin-RevId: 5ac525f010
|
4 years ago |
Megvii Engine Team
|
070c811732
|
fix(imperative): remove convert_inputs
GitOrigin-RevId: a3c43db746
|
3 years ago |
Megvii Engine Team
|
1040b77843
|
fix(mge/functional): fix F.topk(kth_only=True)
GitOrigin-RevId: ddecd1d14b
|
4 years ago |
Megvii Engine Team
|
7c9569e4e5
|
fix(mge/random): fix random seed
GitOrigin-RevId: 121f459b1b
|
3 years ago |
Megvii Engine Team
|
d7b6bfd56c
|
test(mge/fakequant): use fixed input for lsq test to temperarily avoid precision error
GitOrigin-RevId: e91c71874e
|
3 years ago |
Megvii Engine Team
|
5cef74a77e
|
feat(mge/amp): add GradScaler support
GitOrigin-RevId: 0ab4910360
|
4 years ago |
Megvii Engine Team
|
1bf18252c4
|
feat(mge/amp): add mix precision autocast support
GitOrigin-RevId: 6fbffc4845
|
4 years ago |
Megvii Engine Team
|
dedecf6922
|
fix(imperative/utils): fix logical error of replace var
GitOrigin-RevId: 614302552c
|
4 years ago |
Megvii Engine Team
|
43098fb8f1
|
feat(mge): add SlidingWindowTranspose opr
BREAKING CHANGE:
GitOrigin-RevId: 54d726d2fe
|
4 years ago |
Megvii Engine Team
|
df79334cae
|
feat(mge/distributed): add user_pop function to save device memory
BREAKING CHANGE:
GitOrigin-RevId: 0a8e406da5
|
3 years ago |
Megvii Engine Team
|
7225b0f09f
|
fix(mge/utils): use static infer manager to get value of network.varnode
GitOrigin-RevId: ecc47edab8
|
4 years ago |
Megvii Engine Team
|
2d42455fa8
|
fix(mge/utils): fix toposort to get definition order
GitOrigin-RevId: 47a26dd6dd
|
4 years ago |
Megvii Engine Team
|
427113088b
|
fix(module/normalization): fix bug of LayerNorm and support input of any shape
GitOrigin-RevId: fd643addb5
|
4 years ago |
Megvii Engine Team
|
b078dda90b
|
feat(mge/random): add some random op and remove random/distrbution.py
GitOrigin-RevId: 4c05ebc266
|
4 years ago |
Megvii Engine Team
|
f30c0e06a6
|
feat(mgb/opr): add lsq opr
GitOrigin-RevId: 45494a2b57
|
4 years ago |
Megvii Engine Team
|
fea1bba23e
|
fix(mge/tools): fix module status error
GitOrigin-RevId: 38e3eeb34b
|
3 years ago |
Megvii Engine Team
|
6cd01d5a74
|
feat(imperative/functional): let elemwise support empty IO & add some tests
GitOrigin-RevId: a5dc3b997c
|
4 years ago |
Megvii Engine Team
|
0245594156
|
test(autograd): test jvp emulated by 2nd grad
GitOrigin-RevId: 47114fcd99
|
3 years ago |
Megvii Engine Team
|
8480302da8
|
fix(autograd): make higher order grad experimental
GitOrigin-RevId: 81e1eb0ebf
|
4 years ago |
Megvii Engine Team
|
72531f2b4a
|
test(autograd): add more tests for higher order grad
GitOrigin-RevId: 5fc308f87a
|
4 years ago |
Megvii Engine Team
|
522e556b83
|
feat(autodiff): support higher order grad
GitOrigin-RevId: 86390d2179
|
4 years ago |
Megvii Engine Team
|
5198b7834c
|
fix(mge/functional): fix expand_dims for scalar
GitOrigin-RevId: 253ea608f7
|
4 years ago |
Megvii Engine Team
|
933dd9a497
|
feat(mge/distributed): add cuda env check before forked thread
style(core/comp_node): reformat code
GitOrigin-RevId: 372452a8eb
|
4 years ago |
Megvii Engine Team
|
a5060a2bfe
|
feat(mgb/opr): add check_has_inf kernel and opr
GitOrigin-RevId: 0d042dbfce
|
4 years ago |
Megvii Engine Team
|
c64b1c94cd
|
feat(imperative/functional): add roll in functional
GitOrigin-RevId: ff630f1fe7
|
4 years ago |
Megvii Engine Team
|
c78a7848a1
|
test(profiler): simple tests for profiler
GitOrigin-RevId: 92073ce357
|
4 years ago |
Megvii Engine Team
|
54a4d70eb5
|
feat(src/serialization): add support of serializing metadata
GitOrigin-RevId: b563c94451
|
4 years ago |
Megvii Engine Team
|
3e4e4c4604
|
feat(mgb/jit): add graph_opt_config and jit_config interfaces
GitOrigin-RevId: 170d9eeab2
|
4 years ago |
Megvii Engine Team
|
34262d904c
|
fix(imperative): fix the size of blob with offset
GitOrigin-RevId: bb1736dfe1
|
4 years ago |
Megvii Engine Team
|
787f187e7d
|
fix(imperative/src): fix dot backward error
GitOrigin-RevId: 02ba44a0e6
|
4 years ago |
Megvii Engine Team
|
e275dfeca1
|
feat(imperative/python): support pooling mode "average" for avg pool2d module
GitOrigin-RevId: 9fe442129f
|
4 years ago |
Megvii Engine Team
|
d2673c5abf
|
fix(ci/windows): add windows cuda test
GitOrigin-RevId: 706be83032
|
4 years ago |
Megvii Engine Team
|
e417798f9e
|
fix(mge): correct pytype when calling apply from python
GitOrigin-RevId: 6abfa06ada
|
4 years ago |
Megvii Engine Team
|
1525a02530
|
feat(mge/module): add python wrapper for unfold
GitOrigin-RevId: 562103186f
|
4 years ago |
Megvii Engine Team
|
46d96478a1
|
feat(mge/tools): optimize statistical tools
BREAKING CHANGE:
GitOrigin-RevId: cd2a1acd11
|
4 years ago |
Megvii Engine Team
|
7f9d441cbb
|
docs(mge/module): add note about weight shape in conv
GitOrigin-RevId: 43e1f15968
|
4 years ago |
Megvii Engine Team
|
8e5bf94857
|
fix(mge/utils): fix bug of VarNode inplace operations
GitOrigin-RevId: fa9eec7079
|
4 years ago |
Megvii Engine Team
|
13b15fb08c
|
feat(megbrain): add correlation opr
GitOrigin-RevId: 6d44598891
|
4 years ago |
Megvii Engine Team
|
259894250b
|
feat(imperative/autodiff): add grad clip
GitOrigin-RevId: f344f4a233
|
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
|
226f24fa8e
|
test(mge/distributed): add gather scatter reduce broadcast grad test
GitOrigin-RevId: 8245e11f1d
|
4 years ago |