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
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
0b4a767965
feat(mge/distributed): enable uint8 for collective communication
GitOrigin-RevId: 3305c0cf14
3 years ago
Megvii Engine Team
a22b2cf473
ci(copybara): add config files and fix format script
GitOrigin-RevId: 9bd9a7d66c
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
5f558042b2
fix(imperative/ops): use tblgen to generate FastpathCopy
GitOrigin-RevId: b02157dacb
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
f141159088
refactor(mge): loose the error bound of fastrun
GitOrigin-RevId: 9bf9b9d4ca
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
Megvii Engine Team
c2daea3cba
chore(release): bump version
GitOrigin-RevId: a016ea9d56
3 years ago
Megvii Engine Team
207a346351
chore(mge): run get_device_count("gpu") in subprocess
GitOrigin-RevId: 0f0dc001cf
4 years ago
liuke
0baf6b0d63
Merge pull request #175 from tpoisonooo:fix-spell-error
GitOrigin-RevId: fe649676ac
3 years ago
Megvii Engine Team
e6caa9ff89
feat(opr): add bn backward for inference mode
GitOrigin-RevId: bb643cb62f
4 years ago
Xinda Huang
c90fa087ea
test(mge): delete test_external.py
3 years ago
Megvii Engine Team
b2944559a8
fix(imperative/module): remove ``__getattribute__`` method in module
GitOrigin-RevId: 5ac525f010
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