Megvii Engine Team
|
261a5bce23
|
feat(imperative/amp): add dimshuffle in set_format for nhwc
GitOrigin-RevId: 5ced9e1a31
|
3 years ago |
Megvii Engine Team
|
c9e56f4987
|
feat(imperative/amp): add dimshuffle before creating nhwc tensor
GitOrigin-RevId: 4461f9a0d3
|
3 years ago |
Megvii Engine Team
|
cd26376549
|
style(imperative/amp): reformat code
GitOrigin-RevId: 6e5a6e1eaf
|
3 years ago |
Megvii Engine Team
|
3892aa0b6e
|
fix(imperative/amp): fix bn params for nhwc amp
GitOrigin-RevId: 57a3b9d418
|
3 years ago |
Megvii Engine Team
|
6f0b582064
|
chore(imperative/amp): adapt dev
GitOrigin-RevId: 41eb0faadf
|
3 years ago |
Megvii Engine Team
|
c28a875fac
|
fix(imperative/amp): adapt new transformation
GitOrigin-RevId: 6edd577a70
|
3 years ago |
Megvii Engine Team
|
fd41302cc1
|
feat(imperative/amp): add set_format
GitOrigin-RevId: 91de6f49de
|
3 years ago |
Megvii Engine Team
|
fc633ce4ff
|
fix(imperative/amp): fix custom grad in Subgraph
GitOrigin-RevId: 1c728d6ab9
|
3 years ago |
Megvii Engine Team
|
673b295d75
|
feat(imperative/amp): remove conv_format and bn param_dim configs
GitOrigin-RevId: 848d34f63d
|
3 years ago |
Megvii Engine Team
|
fc0f454685
|
fix(dnn/check_non_finite): adjust some details of CheckNonFinite
GitOrigin-RevId: 52ddd805b4
|
3 years ago |
Megvii Engine Team
|
e393d1cf65
|
feat(mge/amp): add convert_format module for NHWC training
GitOrigin-RevId: 1b41e1042c
|
3 years ago |
Megvii Engine Team
|
533fb5bf49
|
feat(imperative): support formatted tensor and add special op rules
GitOrigin-RevId: 77ff909f23
|
3 years ago |
Megvii Engine Team
|
4aa79c453b
|
perf(mge): override grad of matmul
GitOrigin-RevId: d9d97e70fe
|
3 years ago |
Megvii Engine Team
|
98b5ee78c1
|
feat(mge/dnn): add lamb optimizer
GitOrigin-RevId: 5a27157456
|
3 years ago |
Megvii Engine Team
|
a926878c01
|
feat(imperative): remove symbolvar of imperative
GitOrigin-RevId: 16da6d1491
|
3 years ago |
Megvii Engine Team
|
c2e9860feb
|
chore(license): remove all license in file header
GitOrigin-RevId: a0e31247a6
|
3 years ago |
Megvii Engine Team
|
1783b8977a
|
feat(profiler): integrate cupti backend
GitOrigin-RevId: dec8be1908
|
3 years ago |
Megvii Engine Team
|
8182af6eb6
|
fix(mgb): fix strategy of grad_op and opr_attr
GitOrigin-RevId: bb7ab8fa9d
|
3 years ago |
Megvii Engine Team
|
597a1e791b
|
refactor(imperative): add interface to clear algorithm cache
GitOrigin-RevId: 662618954b
|
3 years ago |
Qsingle
|
b4f9703f48
|
fix(overflow): fix the overflow of the long_scalars in network_node and module_stats
|
3 years ago |
Megvii Engine Team
|
dbce6526d6
|
fix(mge/functional): fix return dtype of comparison function
GitOrigin-RevId: 810e32a829
|
3 years ago |
Megvii Engine Team
|
787a22a9d6
|
perf(tensor): implement __new__ in cpp
GitOrigin-RevId: 4defd249c3
|
3 years ago |
Megvii Engine Team
|
53d48ca399
|
fix(fastrun): persistent cache add prefix
(cherry picked from commit bdd82c1976 )
GitOrigin-RevId: 0b13d32175
|
3 years ago |
Megvii Engine Team
|
811579b0e7
|
fix(imperative): restrict value converts to symbolvar
GitOrigin-RevId: 271267be69
|
3 years ago |
Megvii Engine Team
|
409c988163
|
fix(imperative): add matmul apply_on_varnode
GitOrigin-RevId: 2cf6bf237c
|
3 years ago |
Seeker98
|
a22669c132
|
docs(mge/functional): update functional.nn.avg_pool2d docstring
|
3 years ago |
Megvii Engine Team
|
7927e98fd6
|
perf(mge): speed up PixelShuffle
GitOrigin-RevId: 942e755745
|
3 years ago |
Megvii Engine Team
|
43e5f41c4b
|
docs(mgb/docs): add alternative interface for deprecated api
GitOrigin-RevId: fdeaa5b64f
|
3 years ago |
Megvii Engine Team
|
f7e10ea8d7
|
perf(imperative): improve matmul/batch_matmul
GitOrigin-RevId: 4ceb2eb601
|
3 years ago |
Megvii Engine Team
|
1c2a323e78
|
feat(mge): add warning message when mismatched cuda sm is detected
GitOrigin-RevId: f78c79eb06
|
3 years ago |
Megvii Engine Team
|
877bda4180
|
perf(mge): improve cross stream memory borrowing
GitOrigin-RevId: c68977c5dc
|
3 years ago |
Megvii Engine Team
|
84ce94fbd9
|
docs(imperative): fix docs about vision related function
GitOrigin-RevId: 55d12d5e7a
|
3 years ago |
Megvii Engine Team
|
c2435d1561
|
perf(imperative): specialize adaptive pooling
GitOrigin-RevId: 01e1418458
|
3 years ago |
Megvii Engine Team
|
8fcbe8258d
|
docs(mge/functional): fix functional.eye docstring
GitOrigin-RevId: 2645af5e35
|
3 years ago |
Megvii Engine Team
|
d2278f02d2
|
perf(imperative): speed up conv_transpose3d
GitOrigin-RevId: e741305446
|
3 years ago |
Megvii Engine Team
|
3f11b4213a
|
docs(functional): replace megengine function testcode with doctest format
GitOrigin-RevId: f91d5f5afd
|
3 years ago |
Megvii Engine Team
|
60163c5d12
|
docs(functional): replace elem testcode with doctest format
GitOrigin-RevId: a32a011724
|
3 years ago |
Megvii Engine Team
|
fd5031f8eb
|
docs(functional): replace utils testcode with doctest format
GitOrigin-RevId: 8618918ff3
|
3 years ago |
Megvii Engine Team
|
aa63230592
|
docs(functional): replace loss function testcode with doctest format
GitOrigin-RevId: 98224f0e5f
|
3 years ago |
Megvii Engine Team
|
e8d0f9dbf2
|
docs(example): replace dist.help testcode with doctest format
GitOrigin-RevId: 79a7b72ca3
|
3 years ago |
Megvii Engine Team
|
f5f9249a16
|
docs(compose): update compose API docstring and example
GitOrigin-RevId: fd52df2c48
|
3 years ago |
Megvii Engine Team
|
7d3a6db0ef
|
docs(tensor): add more introduction about Tensor
GitOrigin-RevId: 4eba1e6c03
|
3 years ago |
Megvii Engine Team
|
2d72de8a87
|
fix(mge): fix infer output attrs fallible
GitOrigin-RevId: ea18c7f753
|
3 years ago |
Megvii Engine Team
|
fd6f8e58b0
|
feat(mgb/dtype): add dtype qint1
GitOrigin-RevId: abe9fb68b1
|
3 years ago |
Megvii Engine Team
|
f7a5fe17d3
|
feat(mge/config): add a config option for memory forwarding
GitOrigin-RevId: 657326154c
|
3 years ago |
Megvii Engine Team
|
69673f14f7
|
fix(imperative): remove convert_inputs from concat
GitOrigin-RevId: 1511cb4b43
|
3 years ago |
Megvii Engine Team
|
2a900a69cb
|
perf(imperative): improve reduce op performance
GitOrigin-RevId: 26d982a7b8
|
3 years ago |
Megvii Engine Team
|
22be7e11ca
|
docs(mge/functional): typo
GitOrigin-RevId: 0ebc6cb322
|
3 years ago |
Megvii Engine Team
|
111c150db1
|
docs(mge/functional): fix typo
GitOrigin-RevId: 9473f17970
|
3 years ago |
Megvii Engine Team
|
4f60fbbb2f
|
fix(mge/distributed): add polling to solve xmlrpc server io error
GitOrigin-RevId: ca19b1566b
|
3 years ago |