Megvii Engine Team
3eb0505f9b
feat(imperative): add support for quantized conv transpose2d
GitOrigin-RevId: ffd6431299
3 years ago
Megvii Engine Team
dd1fecdf29
feat(mge/opr): add cumsum
GitOrigin-RevId: 740f00a8e5
3 years ago
Megvii Engine Team
7138e4fd02
feat(docs): add docs for megengine.functional.full
GitOrigin-RevId: 8428a0c9a9
3 years ago
Megvii Engine Team
e3fc783642
fix(mgb/opr): fix nvof shape error
fix(mgb/opr): fix format error
fix(mgb/opr): update nvof opr to 1.3 fix (compared to 2.0)
fix(mgb/opr): add try catch to solve test error
GitOrigin-RevId: b688745b36
3 years ago
Megvii Engine Team
3f3a256e0f
fix(mge/functional): fix conv* dtype promotion
GitOrigin-RevId: 3f03790cfc
3 years ago
Megvii Engine Team
536506c3f4
feat(functional): let interpolate support more modes
GitOrigin-RevId: 9693a1ac63
3 years ago
Megvii Engine Team
f76a2cc2c6
feat(mge/opr): add silu and gelu
GitOrigin-RevId: 75aa42947e
3 years ago
Megvii Engine Team
0b764cf2d2
docs(mge/functional): add docs for megengine.functional.full_like
GitOrigin-RevId: 391447e977
3 years ago
Megvii Engine Team
207a346351
chore(mge): run get_device_count("gpu") in subprocess
GitOrigin-RevId: 0f0dc001cf
4 years ago
Megvii Engine Team
070c811732
fix(imperative): remove convert_inputs
GitOrigin-RevId: a3c43db746
3 years ago
Megvii Engine Team
f40df60242
docs(mge): refactor docs to remove warnings
GitOrigin-RevId: efefc2a4a2
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
72ff7aeccb
feat(docs): add docs for megengine.functional.ones_like(jira #MGE-2702)
GitOrigin-RevId: e808fc9e4e
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
75eb04c559
feat(mge/experimental): add WeightScaler support
GitOrigin-RevId: a1d5b68498
4 years ago
Megvii Engine Team
43098fb8f1
feat(mge): add SlidingWindowTranspose opr
BREAKING CHANGE:
GitOrigin-RevId: 54d726d2fe
4 years ago
Megvii Engine Team
ffe2bb2eb2
fix(mge): fix some errors caused by unknown shape when using symbolic trace or building graph
GitOrigin-RevId: 70ddc06eee
4 years ago
Megvii Engine Team
5198b7834c
fix(mge/functional): fix expand_dims for scalar
GitOrigin-RevId: 253ea608f7
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
29f7cdb84a
fix(mgb/opr): correct nvof out shape computation
GitOrigin-RevId: 16bf086e92
4 years ago
Megvii Engine Team
d2673c5abf
fix(ci/windows): add windows cuda test
GitOrigin-RevId: 706be83032
4 years ago
Megvii Engine Team
5e54821845
docs(syncbn): complete syncbn document
GitOrigin-RevId: 9b8d546f0b
4 years ago
Megvii Engine Team
1525a02530
feat(mge/module): add python wrapper for unfold
GitOrigin-RevId: 562103186f
4 years ago
Megvii Engine Team
c6b552cf84
refactor(mge/functional): remove dependence to trace in functional implementations
GitOrigin-RevId: 0b18479fcc
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
8be70d170c
docs(misc): correct docstring format broadly
GitOrigin-RevId: 45234ca07e
4 years ago
Megvii Engine Team
13b15fb08c
feat(megbrain): add correlation opr
GitOrigin-RevId: 6d44598891
4 years ago
Megvii Engine Team
b1bf193e4d
feat(functional/loss): add reduction choices to loss functions
GitOrigin-RevId: a29e6bb4cf
4 years ago
Megvii Engine Team
984d85ca26
feat(mge/functional): argmin and argmax support negtive axis
GitOrigin-RevId: a1bd1102a6
4 years ago
Megvii Engine Team
6972896933
fix(mge/utils): fix network multiple outputs issue
GitOrigin-RevId: d22e639cd3
4 years ago
Megvii Engine Team
e18afa0b05
feat(mge/module): python wrapper for conv_transpose3d
GitOrigin-RevId: 61097b8713
4 years ago
Megvii Engine Team
933188aa93
feat(functional/nn): support F.warp_perspective with `mat_idx`
GitOrigin-RevId: 66910c8bd8
4 years ago
Megvii Engine Team
2d18074a70
fix(mgb): fix spell error
GitOrigin-RevId: acae00e0a5
4 years ago
Megvii Engine Team
65af9cfcdc
refactor(mge): use lower case for default string parameters in functional and module
GitOrigin-RevId: dbc1f27ff7
4 years ago
Megvii Engine Team
d0aa9b41ee
refactor(mge/functional): move nvof to vision, compatible with old usage
GitOrigin-RevId: cd9d9b4f5a
4 years ago
Megvii Engine Team
9dbe71dd1a
feat(mge/utils): add array method for varnode
GitOrigin-RevId: 6e4d05b475
4 years ago
Megvii Engine Team
e9982d613f
chore(mge/functional): add compatible code for functional api
GitOrigin-RevId: 3b2f829cc5
4 years ago
Megvii Engine Team
a3ea1f153c
feat(mgb/opr): add fast profile and combined Execution strategy
GitOrigin-RevId: 843dc3a790
4 years ago
Megvii Engine Team
c8697a7005
feat(imperative/src): python wrapper for cambricon and atlas runtime opr
GitOrigin-RevId: bd969d1339
4 years ago
Megvii Engine Team
8cfed4a1cb
refactor(functional/nn): remove nn.svd
GitOrigin-RevId: 4f58e10a1f
4 years ago
Megvii Engine Team
d970b85deb
feat(functional): add conv3d
GitOrigin-RevId: c06ac77c6c
4 years ago
Megvii Engine Team
46cad4d3c8
feat(functional/ops): add _assert_equal
GitOrigin-RevId: b7ce4158b7
4 years ago
Megvii Engine Team
585aa561f3
fix(functional): replace usage of deprecated get_conv_execution_strategy()
GitOrigin-RevId: 61ef7a50ff
4 years ago
Megvii Engine Team
dcfb6a537e
refactor(mge/functional): move functional api
GitOrigin-RevId: 9cd3e09996
4 years ago
Megvii Engine Team
ae3123b315
feat(mge): add python graph for mgb graph editing
GitOrigin-RevId: 6a9d5beba2
4 years ago
Megvii Engine Team
fe9c6e26a8
feat(imperative/opr): deprecate resize op and make it as a special case of interpolate
GitOrigin-RevId: a5668c5779
4 years ago
Megvii Engine Team
fa4bf16800
feat(mge/functional): add repeat and tile opr
GitOrigin-RevId: a20d4b6fb0
4 years ago
Megvii Engine Team
0c37a588ca
fix(mge/functional): fix F.ones when input is a tensor of scalar type
GitOrigin-RevId: 6d01d6b58d
4 years ago