Megvii Engine Team
|
f2691566fd
|
feat(traced_module): add pytree
GitOrigin-RevId: 6c6e53521c
|
4 years ago |
Megvii Engine Team
|
bee305beb2
|
feat(traced_module): add functional trace and CallMethod/Function expr
GitOrigin-RevId: ad2cdc1b61
|
4 years ago |
Megvii Engine Team
|
763c56f3b9
|
feat(imperative): add traced module
GitOrigin-RevId: 28c3503f2e
|
4 years ago |
Megvii Engine Team
|
9279104b11
|
feat(mge): add opdef serialization and apply_module_trace
GitOrigin-RevId: 5b45bded1d
|
3 years ago |
Megvii Engine Team
|
29d24dbb80
|
fix(mge/function): fix interpolate unsupport fp16 error
GitOrigin-RevId: 7fc6271986
|
3 years ago |
Megvii Engine Team
|
e21967bb40
|
feat(mge): add env MGE_FASTRUN_CACHE_DIR
GitOrigin-RevId: 0351ff88d1
|
3 years ago |
Megvii Engine Team
|
6a1ec8a890
|
feat(mge): add git commit-id into fastrun cache key
GitOrigin-RevId: fb614b4d9c
|
3 years ago |
Megvii Engine Team
|
ae87876d34
|
feat(mge): refactor weightscaler
GitOrigin-RevId: 7f874388f7
|
3 years ago |
Megvii Engine Team
|
67575d582c
|
feat(mge/opr): add interpolate bilinear mode
GitOrigin-RevId: f7023a3fd3
|
3 years ago |
Megvii Engine Team
|
0558b2123d
|
feat(mge/opr): add interpolate nearest mode
GitOrigin-RevId: d384b87f50
|
3 years ago |
Megvii Engine Team
|
d86ed426ee
|
fix(dtr): simulate the system stack to avoid stack overflow during recomputing
GitOrigin-RevId: cb73e62b19
|
3 years ago |
Megvii Engine Team
|
12cdbddd14
|
fix(ci): clean fastrun cache in windows and macos ci
GitOrigin-RevId: d1a010287f
|
3 years ago |
Megvii Engine Team
|
31705913c0
|
fix(ci): set MGE_FASTRUN_CACHE_TYPE=FILE in ci env
GitOrigin-RevId: c4a549480e
|
3 years ago |
Megvii Engine Team
|
3c6665f7c1
|
feat(lite/whl): merge lite whl to main package
GitOrigin-RevId: 27c7e50207
|
3 years ago |
Megvii Engine Team
|
989fdde255
|
refactor(subgraph): use graph queue to cache compiled op graphs
GitOrigin-RevId: cba8574c73
|
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
|
3206af9db2
|
perf(functional/matmul): reimplement matmul with subgraph
GitOrigin-RevId: 456b2a51d3
|
3 years ago |
Megvii Engine Team
|
8c47c1f149
|
perf(syncbn): reimplement with subgraph
GitOrigin-RevId: 13e7e3d3c0
|
3 years ago |
Megvii Engine Team
|
53da5c79f4
|
feat(cg): add comp_seq_sync_device option
GitOrigin-RevId: c2199c59e9
|
3 years ago |
Megvii Engine Team
|
e1c7b22ff0
|
perf(ops): enable memory forward for reduce in special cases
GitOrigin-RevId: dd6e1664c5
|
3 years ago |
Megvii Engine Team
|
cd60d26852
|
perf(ops): specialize Broadcast
GitOrigin-RevId: 0cba3e6e93
|
3 years ago |
Megvii Engine Team
|
3fd3e000d1
|
feat(ops): add serval utility ops
GitOrigin-RevId: 623cb5ddfc
|
3 years ago |
Megvii Engine Team
|
5b4f7c5dd0
|
perf(interpreter): unwind ops with make_forward_graph
GitOrigin-RevId: 5fb8c85089
|
3 years ago |
Megvii Engine Team
|
5798f6ce20
|
feat(subgraph): add OpMeth make_forward_graph
GitOrigin-RevId: 171301fc2b
|
3 years ago |
Megvii Engine Team
|
48db45d123
|
perf(interpreter): try put device value with host to reduce d2h
GitOrigin-RevId: 63d36e7706
|
3 years ago |
Megvii Engine Team
|
a605f38b26
|
refactor(opmeth): add OpMethCache struct
GitOrigin-RevId: c1ebe15672
|
3 years ago |
Megvii Engine Team
|
0213dbe556
|
feat(subgraph): add graph builder
GitOrigin-RevId: f32cfc39e0
|
3 years ago |
Megvii Engine Team
|
0b8dc2c98b
|
refactor(subgraph): add generic encoded_graph
GitOrigin-RevId: 56d90be0e7
|
3 years ago |
Megvii Engine Team
|
88b3c84229
|
refactor(subgraph): move to subgraph.h
GitOrigin-RevId: 2791f335d4
|
3 years ago |
Megvii Engine Team
|
1e3af4dd17
|
fix(mgb/comp_node): add more info in `comp_node.to_string()`
GitOrigin-RevId: 794a8847aa
|
3 years ago |
Megvii Engine Team
|
3103180456
|
fix(mge/optimizer): only disable convert inputs in build-in optimizers
GitOrigin-RevId: 1a48fe318d
|
3 years ago |
Megvii Engine Team
|
fd24dc8e36
|
feat(mge/hub/fetcher): add checking of mge version
GitOrigin-RevId: 808af7f82a
|
3 years ago |
Megvii Engine Team
|
0ef5183c65
|
perf(opdef/reshape): specialize Reshape
GitOrigin-RevId: 26d0e151ca
|
3 years ago |
Megvii Engine Team
|
77309609fa
|
perf(functional/dropout): add fastpath for dropout
GitOrigin-RevId: 3bf8546908
|
3 years ago |
Megvii Engine Team
|
92079e1061
|
docs(mge/distributed): update distributed collective communication Google style docs
GitOrigin-RevId: 871c4052b0
|
3 years ago |
Megvii Engine Team
|
5fe789abb9
|
fix(mge/distributed): skip check when machine_ranks is None
GitOrigin-RevId: a18ee052e9
|
3 years ago |
Megvii Engine Team
|
298c44433f
|
fix(mge/module): fix mistake warning message of module auto naming
GitOrigin-RevId: 4b85c13e90
|
3 years ago |
Megvii Engine Team
|
c54a731b9b
|
fix(utils/network): fix replace oprs
GitOrigin-RevId: eba27e3dfb
|
3 years ago |
Megvii Engine Team
|
ebb7f5cac2
|
fix(imperative/python/test): add device judgment for the test case of conv-transpose2d-quantized
GitOrigin-RevId: 456cb78439
|
3 years ago |
Megvii Engine Team
|
2950dd8d69
|
test(imperative): test autodiff.Function with non tensor arguments
GitOrigin-RevId: 6114f48d21
|
3 years ago |
Megvii Engine Team
|
20e8541bbb
|
refactor(imperative): bind fallback impl on first op method call
GitOrigin-RevId: 82ae1e3205
|
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
|
3e58cbb8d2
|
fix(mge/quantization): fix lsq init format error
GitOrigin-RevId: 032de9c773
|
3 years ago |
Megvii Engine Team
|
b3d2a632f3
|
fix(sublinear): fix lb_memory in sublinear
GitOrigin-RevId: fd51023c01
|
3 years ago |
Megvii Engine Team
|
b74afde89e
|
feat(mgb/opr): let reduce support empty IO
GitOrigin-RevId: 88b37123a8
|
3 years ago |
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
|
dd1fecdf29
|
feat(mge/opr): add cumsum
GitOrigin-RevId: 740f00a8e5
|
3 years ago |