Megvii Engine Team
cc6084b4d5
revert: chore(mge/misc): api converage
This reverts commit --
77fd432cb2
0d7126641b
559d205a8c
6d8e7398e4
b72517bfe9
GitOrigin-RevId: 18dc4ce999
2 years ago
Megvii Engine Team
98460f583f
chore(mge/misc): rename APIs that should not be public
GitOrigin-RevId: b72517bfe9
2 years ago
Megvii Engine Team
b8c7557b86
fix(mm): fix mm error when use sync
GitOrigin-RevId: 63387bda04
2 years ago
Megvii Engine Team
ac51f7807d
feat(mge/distributed): add support for batch send recv op
GitOrigin-RevId: eb3d712704
3 years ago
Megvii Engine Team
c2e9860feb
chore(license): remove all license in file header
GitOrigin-RevId: a0e31247a6
3 years ago
Megvii Engine Team
e8d0f9dbf2
docs(example): replace dist.help testcode with doctest format
GitOrigin-RevId: 79a7b72ca3
3 years ago
Megvii Engine Team
4f60fbbb2f
fix(mge/distributed): add polling to solve xmlrpc server io error
GitOrigin-RevId: ca19b1566b
3 years ago
Megvii Engine Team
0bdd0b1467
refactor(dispatch): switch to new dispatch system
GitOrigin-RevId: 32dd49a23a
3 years ago
Megvii Engine Team
6102b96ca1
Merge pull request #407 from ShrimpLau:docstring_all_reduce_xxx
GitOrigin-RevId: 3d15b14fd1
3 years ago
quqi.liu
0f5c26d746
docs(mge/distributed): change the allreduce related functions document
3 years ago
quqi.liu
da8b16fc9d
docs(mge/distributed): change the allreduce related functions document
3 years ago
Megvii Engine Team
05ee603843
fix(mge/distributed): fix single_machine_cache error
GitOrigin-RevId: be57111408
3 years ago
Megvii Engine Team
dff7719e87
feat(mge/distributed): add preload host data with op
fix(mge/distributed): change api name with preload
fix(mge/distributed): fix recursive model in preload tensor
fix(mge/distributed): fix recursive when cache contain None
GitOrigin-RevId: 80e2a6dd70
3 years ago
Megvii Engine Team
c5f8b58385
docs(docstring): transfer to google style
GitOrigin-RevId: a71245c553
3 years ago
Megvii Engine Team
03e80759b6
feat(mge/distributed): add hybird parallel Opr
GitOrigin-RevId: ff26671746
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
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
f2ac4c345b
docs(distributed.functional.all_reduce_sum): googlestring and examples
GitOrigin-RevId: a456dfde24
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
207a346351
chore(mge): run get_device_count("gpu") in subprocess
GitOrigin-RevId: 0f0dc001cf
4 years ago
Megvii Engine Team
551cc701c6
docs(distributed.functional): add return type for all_reduce_max (jira #MGE-2706)
GitOrigin-RevId: a29f1f1880
3 years ago
Megvii Engine Team
497ef6c337
fix(mge/dist): fix gl oom error
GitOrigin-RevId: 4ba3d2cfd7
3 years ago
Megvii Engine Team
df79334cae
feat(mge/distributed): add user_pop function to save device memory
BREAKING CHANGE:
GitOrigin-RevId: 0a8e406da5
3 years ago
Megvii Engine Team
809d5056cd
feat(mge/distributed): enable pt shm allreduce
GitOrigin-RevId: 1dd5a02a51
4 years ago
Megvii Engine Team
933dd9a497
feat(mge/distributed): add cuda env check before forked thread
style(core/comp_node): reformat code
GitOrigin-RevId: 372452a8eb
4 years ago
Megvii Engine Team
aa4e847692
fix(interpreter): release gil when interpreter blocking
GitOrigin-RevId: c48e9efa5b
4 years ago
Megvii Engine Team
90d3905799
feat(mge): add mge._exit
GitOrigin-RevId: e7f348a677
4 years ago
Megvii Engine Team
0cf4ff702e
perf(dist): add fastpath for bcast params
GitOrigin-RevId: aa40b3cd72
4 years ago
Megvii Engine Team
a7581986bc
perf(dist): speed up bcast_val
GitOrigin-RevId: 21c4123b09
4 years ago
Megvii Engine Team
aea5ac1363
fix(mge/distributed): fix gather scatter reduce broadcast autodiff
GitOrigin-RevId: 1c2250a079
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
1a7112997c
feat(opr-mm): add backend argument for remote send/recv
GitOrigin-RevId: 841a0e45ab
4 years ago
Megvii Engine Team
dee5a10af9
feat(distributed): auto detect device and backend when init group
GitOrigin-RevId: 90be2d5b4d
4 years ago
Megvii Engine Team
dcff115ef8
fix(distributed/launcher): fetch data early to avoid dead lock in launcher
GitOrigin-RevId: 9abcc956ef
4 years ago
Megvii Engine Team
dcfb6a537e
refactor(mge/functional): move functional api
GitOrigin-RevId: 9cd3e09996
4 years ago
Megvii Engine Team
fa4bf16800
feat(mge/functional): add repeat and tile opr
GitOrigin-RevId: a20d4b6fb0
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
17323dbd73
feat(dist): collect return values in dist.launcher
GitOrigin-RevId: 519e768ce9
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
4 years ago
Megvii Engine Team
10d9d77e41
refactor(mge/distributed): remove get_free_ports func
GitOrigin-RevId: b85c4885be
4 years ago
Megvii Engine Team
0adf49b137
fix(mge/distributed): fix deadlock by mixing thread and fork
GitOrigin-RevId: c138cb9c28
4 years ago
Megvii Engine Team
ea8eb4cf72
feat(mge/distributed): scalar support for distributed functions
GitOrigin-RevId: 53f3575baf
4 years ago
Megvii Engine Team
d3bfb0e983
fix(mge): fix trace exit code and reformat
GitOrigin-RevId: 145c06b7e7
4 years ago
Megvii Engine Team
de0742be25
refactor(mge): reopen passed assertions
GitOrigin-RevId: e0276e73e3
4 years ago
Megvii Engine Team
a90c937d22
feat(interpreter): add command buffer for inplace
GitOrigin-RevId: 020d1e88d4
4 years ago
Megvii Engine Team
75129cf071
chore(mge): clean up before merge to dev
* remove dead test
* clean some codes
* fix test_fake_quant due to change of implementation
GitOrigin-RevId: f030a9966d
4 years ago
Megvii Engine Team
b9762d714c
fix(mge): make parampack run with tensor symbolic shape
GitOrigin-RevId: 6fc313785d
4 years ago