Xinran Xu
|
0b1d516e38
|
chore(release): bump version
|
5 years ago |
Megvii Engine Team
|
f0959dc9c4
|
refactor(mge/examples): add example to use modified sublinear API
GitOrigin-RevId: c6f6c65d0b
|
5 years ago |
Megvii Engine Team
|
a2f0e8788f
|
feat(mge/api): expose sublinear related parameters at mge api level
GitOrigin-RevId: 7a47f5d0d5
|
5 years ago |
Xinran Xu
|
f44d2632da
|
chore(release): bump version
|
5 years ago |
Megvii Engine Team
|
d84dc7f75c
|
build(python_module): use consistent flag declaration in SWIG and c++
Include cmake generated megbrain_build_config.h in SWIG in order to remove usage of MGB_DEF.
GitOrigin-RevId: d3e841eaf6
|
5 years ago |
Megvii Engine Team
|
03728d45da
|
fix(mgb/build): fix multi-machine macro and add test_distributed
GitOrigin-RevId: cb1bfe8742
|
5 years ago |
Megvii Engine Team
|
d74a4ee931
|
fix(mge/build): use consistent genfiles/megbrain_build_config.h
Modify get_dynamic_info to work with cmake changes
Also rename get_dynamic_info as get_mem_allocation_info
GitOrigin-RevId: 383affe388
|
5 years ago |
Megvii Engine Team
|
720b6b49d6
|
build(mge): copy file required by setup.py to build directory
GitOrigin-RevId: 0b853e809f
|
5 years ago |
Megvii Engine Team
|
e8494f729b
|
fix(unittest): remove the torch dependency in test_conv
GitOrigin-RevId: 36dcab2b95
|
5 years ago |
Megvii Engine Team
|
779ed45d74
|
docs(mge/parampack): refine parampack doc
GitOrigin-RevId: 1cff83c308
|
5 years ago |
Megvii Engine Team
|
8e8794040d
|
chore(release): bump version.
GitOrigin-RevId: 979d8c247b
|
5 years ago |
Megvii Engine Team
|
7ed98dd50a
|
feat(mge/functional): add conv_transpose2d support
GitOrigin-RevId: 9ad87a4ea9
|
5 years ago |
Megvii Engine Team
|
20332705d3
|
docs(mge/functional): refine tensor transpose doc
GitOrigin-RevId: 763d74d61e
|
5 years ago |
Megvii Engine Team
|
044695aa2a
|
feat(mgb/plugin): add profiler for dynamic graph
GitOrigin-RevId: 10a6cccff9
|
5 years ago |
Megvii Engine Team
|
b588d93eb0
|
fix(mge): fix fast-run cache mismatch
GitOrigin-RevId: a41150f71a
|
5 years ago |
Megvii Engine Team
|
0bdb64c5f9
|
ci(test): dump test time result to json file
GitOrigin-RevId: 0ed092a0e8
|
5 years ago |
Megvii Engine Team
|
dbe2b89331
|
fix(midout): fix brain opr midout 2/2 (also see a6aa1574)
fix extern_c_opr midout
GitOrigin-RevId: 7de4f650d1
|
5 years ago |
Megvii Engine Team
|
888895e971
|
refactor(mge/test): make test_parampack more stable
GitOrigin-RevId: d82230ea07
|
5 years ago |
Megvii Engine Team
|
da52256891
|
fix(mge/data): process mnist without generate new files
GitOrigin-RevId: 44a697c3fe
|
5 years ago |
Megvii Engine Team
|
5178648422
|
fix(mge/doc): fix batch_norm2d doc
GitOrigin-RevId: 769ddd367b
|
5 years ago |
Megvii Engine Team
|
f077a5292c
|
fix(mge/param_pack): release old parameters
GitOrigin-RevId: 40a1f044e9
|
5 years ago |
Megvii Engine Team
|
48822557b8
|
docs(mge/bn): fix docs and tests of batchnorm
GitOrigin-RevId: 8a96aa5fc2
|
5 years ago |
Megvii Engine Team
|
2c4d1afe14
|
test(mge): move `NUMBER` config to pytest.init
GitOrigin-RevId: 1d82209c40
|
5 years ago |
Megvii Engine Team
|
f33a92d635
|
docs(mge): fix api docs of loss and tranform
GitOrigin-RevId: 4d4d32b3c4
|
5 years ago |
Megvii Engine Team
|
fd83e0252d
|
fix(mge/test): fix reproducibility of test_correctness.py
GitOrigin-RevId: 70bd43bbab
|
5 years ago |
Megvii Engine Team
|
e1b2d31d36
|
test(mge): skip some doctests having different results on gpu
GitOrigin-RevId: 66e5db5c89
|
5 years ago |
Megvii Engine Team
|
20b67b294c
|
ci(python): add format and lint check for python
GitOrigin-RevId: d8d106b927
|
5 years ago |
Megvii Engine Team
|
62b0c6cd27
|
style(mge): apply format.sh
GitOrigin-RevId: a900b1bb6e
|
5 years ago |
Megvii Engine Team
|
fc6aa12e2f
|
feat(mge):add qint4/quint4 to python
GitOrigin-RevId: f94609db00
|
5 years ago |
Megvii Engine Team
|
2c2caf3331
|
refactor(mge/data/dataloader): refactor the implementation of parallel dataloader
GitOrigin-RevId: 0554ee8427
|
5 years ago |
Megvii Engine Team
|
364dafccf6
|
fix typo
GitOrigin-RevId: e1c3287467
|
5 years ago |
Megvii Engine Team
|
94a7696924
|
refactor ImageNet
GitOrigin-RevId: f7774e0ffc
|
5 years ago |
Megvii Engine Team
|
e677ffc85d
|
fix(mge/dataset) fix imagenet dir error
GitOrigin-RevId: e0a59464bb
|
5 years ago |
Megvii Engine Team
|
c9830d20d0
|
fix(mge/test): fix tolerance
GitOrigin-RevId: 58c029b394
|
5 years ago |
Megvii Engine Team
|
2174fd12b9
|
refactor(mge/test): make test_converge more stable
GitOrigin-RevId: 710e3ede40
|
5 years ago |
Megvii Engine Team
|
dd41e2e47a
|
fix(mge/test):fix integration/manual tests
GitOrigin-RevId: 5933392407
|
5 years ago |
Megvii Engine Team
|
8d2bbf7383
|
fix(mge/functional): support scalar inputs in elemwise functions
GitOrigin-RevId: 7bce561ee1
|
5 years ago |
Megvii Engine Team
|
ea00b57d93
|
refactor(mge/module): use mge rng to sample from uniform/normal distribution
GitOrigin-RevId: 6ec8f99af5
|
5 years ago |
Megvii Engine Team
|
a1597cfc46
|
docs(mge/param_pack): add description of param pack and add param pack in example run_dist
GitOrigin-RevId: 1c900e159d
|
5 years ago |
Megvii Engine Team
|
646829b802
|
ci(pytest): use common method install pip package
GitOrigin-RevId: 87162b8b56
|
5 years ago |
Megvii Engine Team
|
9edce7e294
|
ci(python_test): run python test with cuda
GitOrigin-RevId: bc51c2e4d5
|
5 years ago |
Megvii Engine Team
|
80d3a06f9e
|
fix(mge/param_pack): fix param pack with just one param
GitOrigin-RevId: 0d28a12e59
|
5 years ago |
Megvii Engine Team
|
876e799ed2
|
add(mge) add internal check for mge
GitOrigin-RevId: 7f293c9d03
|
5 years ago |
Megvii Engine Team
|
a744b3cb8a
|
feat(mge/module):add param pack
GitOrigin-RevId: 9cf1dbe44d
|
5 years ago |
Megvii Engine Team
|
afcda610f9
|
fix(mge/data/dataloader): fix typo, import and refine the logic of plasma_store
GitOrigin-RevId: 7d169a5294
|
5 years ago |
Megvii Engine Team
|
65432d3b93
|
fix(mge/module): fix torch subgraph under jit.trace with symbolic=False
GitOrigin-RevId: a208ba79d9
|
5 years ago |
Megvii Engine Team
|
862de28a64
|
build(mgb): fix linkage of libatomic
GitOrigin-RevId: c164ed28e1
|
5 years ago |
Megvii Engine Team
|
d4f6a7bfed
|
fix(mge): fix tensor docstring
GitOrigin-RevId: 4dab166976
|
5 years ago |
Megvii Engine Team
|
e6d7c5e99a
|
docs(mge): fix docstring in loss and dataset
GitOrigin-RevId: 6b56673415
|
5 years ago |
Megvii Engine Team
|
f91881ffdc
|
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
|
5 years ago |