Megvii Engine Team
e918f0aa75
feat(traced_module): add treedef leaf node check and add some graph api
GitOrigin-RevId: 36c069bfee
3 years ago
Megvii Engine Team
c7e730bc12
feat(traced_module): add some functions of graph modification
GitOrigin-RevId: 09691ebd33
3 years ago
Megvii Engine Team
f88bd3ae33
refactor(traced_module): let TracedModule own argdef_graph_map
GitOrigin-RevId: 80d685b9a3
3 years ago
Megvii Engine Team
b1c46ba46c
feat(traced_module): add some functions of graph modification
GitOrigin-RevId: ac0603057a
3 years ago
Megvii Engine Team
4bb253695b
feat(traced_module): let CallFunction own graph
GitOrigin-RevId: 66cdbca7e5
4 years ago
Megvii Engine Team
9a6a379346
feat(traced_module): add visit method
GitOrigin-RevId: 251ecebf87
4 years ago
Megvii Engine Team
442b4f6c26
test(traced_module): add some testcases for traced module
GitOrigin-RevId: 0d6bb20b2b
4 years ago
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
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
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
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
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
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
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
d16c5caf6b
fix(mge/dump): fix dump device error with const
GitOrigin-RevId: 9dd8321fd7
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
d811dc5478
docs(mge/distributed): add document for distributed.backend
GitOrigin-RevId: 6cdcf7af77
3 years ago
Megvii Engine Team
9526ee521b
docs(distributed.functional): add return type for all_reduce_min
GitOrigin-RevId: 9f734902fe
3 years ago
Megvii Engine Team
f76a2cc2c6
feat(mge/opr): add silu and gelu
GitOrigin-RevId: 75aa42947e
3 years ago
Megvii Engine Team
f2ac4c345b
docs(distributed.functional.all_reduce_sum): googlestring and examples
GitOrigin-RevId: a456dfde24
3 years ago
Megvii Engine Team
186bacfb71
fix(mge): recover bn freeze fastpath execution
GitOrigin-RevId: 13e58c9fba
3 years ago
Megvii Engine Team
bfc4e7a966
docs(mge): fix amp docstring problems
GitOrigin-RevId: e5540eb940
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
1f0436967c
refactor(mge/distributed): using nccl as default in distributed training
GitOrigin-RevId: 81268e84bc
3 years ago
Megvii Engine Team
b17a02d44a
feat(mge/distributed): deprecate get_device_count_by_fork
GitOrigin-RevId: 6147c3ae90
3 years ago