Megvii Engine Team
|
de8ffe0c12
|
refactor(imperative): unify interpreter option setting
GitOrigin-RevId: 53510445cc
|
3 years ago |
Megvii Engine Team
|
68cde8734e
|
fix(mge/imperative): support broadcast with None
GitOrigin-RevId: dd330a2a1d
|
3 years ago |
Megvii Engine Team
|
0bdd0b1467
|
refactor(dispatch): switch to new dispatch system
GitOrigin-RevId: 32dd49a23a
|
3 years ago |
Megvii Engine Team
|
a72e0cb568
|
feat(imperative,src): add jit builder for custom op
GitOrigin-RevId: 3bb0b46311
|
3 years ago |
Megvii Engine Team
|
7fa70df9f2
|
style(imperative): remove absolute imports in megengine
GitOrigin-RevId: 7004929069
|
3 years ago |
Megvii Engine Team
|
b75d10094d
|
fix(mge/elwise): removed back to fp32 mode
GitOrigin-RevId: 6d1720f4af
|
3 years ago |
Megvii Engine Team
|
9fd2e66350
|
feat(mge/elwise): removed back to fp32 mode
GitOrigin-RevId: a665a279a6
|
3 years ago |
Megvii Engine Team
|
8a222d2c9c
|
fix(mge): replace functional paramter
GitOrigin-RevId: af3c68c588
|
3 years ago |
Megvii Engine Team
|
27aa648ba7
|
fix(mge): control compute_mode by context
GitOrigin-RevId: f3968309f4
|
3 years ago |
Megvii Engine Team
|
a404c508e9
|
feat(mge): support dump with specific format
GitOrigin-RevId: 57a7c0de02
|
3 years ago |
Megvii Engine Team
|
cebda6ff8d
|
feat(mge/imperative): add ctc loss
GitOrigin-RevId: e29854a98e
|
3 years ago |
Megvii Engine Team
|
cdb692d2fa
|
refactor(imperative): add TODO tag for some functions
GitOrigin-RevId: e295a1fa55
|
3 years ago |
Megvii Engine Team
|
90dd07161c
|
refactor(imperative): modify the python interface of custom op
GitOrigin-RevId: e82e5de480
|
3 years ago |
Megvii Engine Team
|
cbf024bf20
|
feat(imperative): add adaptor between custom op and imperative runtime
GitOrigin-RevId: d7877f2e32
|
3 years ago |
Megvii Engine Team
|
27f2ecc436
|
feat(imperative/opr): add extern c opr and support dump
GitOrigin-RevId: b29e374c60
|
3 years ago |
Megvii Engine Team
|
c5f8b58385
|
docs(docstring): transfer to google style
GitOrigin-RevId: a71245c553
|
3 years ago |
Megvii Engine Team
|
5431929ec7
|
feat(functional): let advance indexing support empty tensor and add more tests
GitOrigin-RevId: 49e1492934
|
3 years ago |
Megvii Engine Team
|
a7a3bf2d6c
|
test(subgraph): simple test for subgraph
GitOrigin-RevId: 3d6ecd5db7
|
3 years ago |
Megvii Engine Team
|
d063d5774f
|
perf(functional): use fma to reduce elemwise but disable subgraph compilation
GitOrigin-RevId: c75a6e1a09
|
3 years ago |
Megvii Engine Team
|
2a063f8e87
|
fix(subgraph): fix scope mismatch of subgraph content
GitOrigin-RevId: 6e23456250
|
3 years ago |
Megvii Engine Team
|
8c47c1f149
|
perf(syncbn): reimplement with subgraph
GitOrigin-RevId: 13e7e3d3c0
|
3 years ago |
Megvii Engine Team
|
18274e023c
|
fix(imperative): fix error message when applying custom function with non-tensor arguments
GitOrigin-RevId: 387d6fda4a
|
3 years ago |
Megvii Engine Team
|
bfc4e7a966
|
docs(mge): fix amp docstring problems
GitOrigin-RevId: e5540eb940
|
3 years ago |
Megvii Engine Team
|
070c811732
|
fix(imperative): remove convert_inputs
GitOrigin-RevId: a3c43db746
|
3 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
|
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
|
2cd9823210
|
fix(mgb/tensorrt): fix trt runtime, padding channel to a multiple of 4 when using kCHW4 IOFormat
GitOrigin-RevId: c5f1ed70da
|
3 years ago |
Megvii Engine Team
|
522e556b83
|
feat(autodiff): support higher order grad
GitOrigin-RevId: 86390d2179
|
4 years ago |
Megvii Engine Team
|
54a4d70eb5
|
feat(src/serialization): add support of serializing metadata
GitOrigin-RevId: b563c94451
|
4 years ago |
Megvii Engine Team
|
241b35a697
|
refactor(ops): remove BackwardGraph op
GitOrigin-RevId: eda20e5760
|
4 years ago |
Megvii Engine Team
|
8be70d170c
|
docs(misc): correct docstring format broadly
GitOrigin-RevId: 45234ca07e
|
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
|
87d6ff228d
|
fix(mge): fix dumping backward graph
GitOrigin-RevId: 430f110053
|
4 years ago |
Megvii Engine Team
|
6972896933
|
fix(mge/utils): fix network multiple outputs issue
GitOrigin-RevId: d22e639cd3
|
4 years ago |
Megvii Engine Team
|
13e6ea349d
|
feat(imperative/opr): rebase rng refactoring to dev & add python module
GitOrigin-RevId: ee5984c52d
|
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
|
9dbe71dd1a
|
feat(mge/utils): add array method for varnode
GitOrigin-RevId: 6e4d05b475
|
4 years ago |
Megvii Engine Team
|
5a7c30e0f7
|
feat(mge/tools): add summary print for module_stats and network_visualize
GitOrigin-RevId: 7d85aa0ea2
|
4 years ago |
Megvii Engine Team
|
702ed9eee6
|
feat(mge): enable cg dump for debugging
GitOrigin-RevId: 4292364d0d
|
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
|
e6b06914f7
|
feat(mge/tensor): support non-Tensor value in `_reset` and remove depreciated tests
GitOrigin-RevId: faf6c78aa8
|
4 years ago |
Megvii Engine Team
|
594fa722bd
|
feat(mge/graph): add `modify_opr_algo_strategy_inplace` for fast-run
GitOrigin-RevId: 034cf58b2a
|
4 years ago |
Megvii Engine Team
|
813be72ce9
|
feat(mge/utils): refactor GraphInference and add more options
GitOrigin-RevId: 44b96dbf3d
|
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
|
fa4bf16800
|
feat(mge/functional): add repeat and tile opr
GitOrigin-RevId: a20d4b6fb0
|
4 years ago |
Megvii Engine Team
|
1d7dd00144
|
feat(mge/quantization): add QParams and QuantDtypeMeta for quantization data structure
GitOrigin-RevId: df3416fe13
|
4 years ago |
Megvii Engine Team
|
0a191f8d48
|
fix(misc): some fixes to make faster_rcnn run in no-symbolic-trace
GitOrigin-RevId: 0fc40ad9c8
|
4 years ago |
Megvii Engine Team
|
4f3875eb4f
|
docs(mge): add missing docstring and fix sphinx build warnings
GitOrigin-RevId: 4ce73cfd80
|
4 years ago |
Megvii Engine Team
|
5b697c714c
|
chore(imperative): remove unused reference to megbrain_graph
GitOrigin-RevId: 2c31f13bb8
|
4 years ago |