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 |
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
|
72ff7aeccb
|
feat(docs): add docs for megengine.functional.ones_like(jira #MGE-2702)
GitOrigin-RevId: e808fc9e4e
|
3 years ago |
Megvii Engine Team
|
7c9569e4e5
|
fix(mge/random): fix random seed
GitOrigin-RevId: 121f459b1b
|
3 years ago |
Megvii Engine Team
|
d7b6bfd56c
|
test(mge/fakequant): use fixed input for lsq test to temperarily avoid precision error
GitOrigin-RevId: e91c71874e
|
3 years ago |
Megvii Engine Team
|
5cef74a77e
|
feat(mge/amp): add GradScaler support
GitOrigin-RevId: 0ab4910360
|
4 years ago |
Megvii Engine Team
|
1bf18252c4
|
feat(mge/amp): add mix precision autocast support
GitOrigin-RevId: 6fbffc4845
|
4 years ago |
Megvii Engine Team
|
f12355f727
|
fix(imperative/grad): fix hardcode dtype in subtensor_grad_rule
GitOrigin-RevId: 50da4af26d
|
4 years ago |
Megvii Engine Team
|
56b94d89a2
|
feat(dtr): add sqrt sampling
GitOrigin-RevId: 8cb2ceb520
|
3 years ago |
Megvii Engine Team
|
8a73193c2d
|
feat(dtr): remove eviction threshold
GitOrigin-RevId: 35c2014bf3
|
3 years ago |
Megvii Engine Team
|
75eb04c559
|
feat(mge/experimental): add WeightScaler support
GitOrigin-RevId: a1d5b68498
|
4 years ago |
Megvii Engine Team
|
dedecf6922
|
fix(imperative/utils): fix logical error of replace var
GitOrigin-RevId: 614302552c
|
4 years ago |
Megvii Engine Team
|
497ef6c337
|
fix(mge/dist): fix gl oom error
GitOrigin-RevId: 4ba3d2cfd7
|
3 years ago |
Megvii Engine Team
|
43098fb8f1
|
feat(mge): add SlidingWindowTranspose opr
BREAKING CHANGE:
GitOrigin-RevId: 54d726d2fe
|
4 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
|
7225b0f09f
|
fix(mge/utils): use static infer manager to get value of network.varnode
GitOrigin-RevId: ecc47edab8
|
4 years ago |
Megvii Engine Team
|
ffe2bb2eb2
|
fix(mge): fix some errors caused by unknown shape when using symbolic trace or building graph
GitOrigin-RevId: 70ddc06eee
|
4 years ago |
Megvii Engine Team
|
2d42455fa8
|
fix(mge/utils): fix toposort to get definition order
GitOrigin-RevId: 47a26dd6dd
|
4 years ago |
Megvii Engine Team
|
0c97b2a3ec
|
fix(module): remove assert during forward
GitOrigin-RevId: bb04d861a3
|
3 years ago |
Megvii Engine Team
|
427113088b
|
fix(module/normalization): fix bug of LayerNorm and support input of any shape
GitOrigin-RevId: fd643addb5
|
4 years ago |
Megvii Engine Team
|
a95f6d4f75
|
perf(trace): add fastpath for const value assert
GitOrigin-RevId: 9a966f257f
|
3 years ago |
Megvii Engine Team
|
2cd9823210
|
fix(mgb/tensorrt): fix trt runtime, padding channel to a multiple of 4 when using kCHW4 IOFormat
GitOrigin-RevId: c5f1ed70da
|
3 years ago |
Megvii Engine Team
|
b078dda90b
|
feat(mge/random): add some random op and remove random/distrbution.py
GitOrigin-RevId: 4c05ebc266
|
4 years ago |
Megvii Engine Team
|
f30c0e06a6
|
feat(mgb/opr): add lsq opr
GitOrigin-RevId: 45494a2b57
|
4 years ago |
Megvii Engine Team
|
eb66681f45
|
fix(mge/random): fix delete_rng_handel
GitOrigin-RevId: a836e13c23
|
3 years ago |
Megvii Engine Team
|
d6db4fea93
|
fix(mge/module): set no_cache=true when loading state dict
GitOrigin-RevId: 83281a3d47
|
4 years ago |
Megvii Engine Team
|
fea1bba23e
|
fix(mge/tools): fix module status error
GitOrigin-RevId: 38e3eeb34b
|
3 years ago |
Megvii Engine Team
|
6cd01d5a74
|
feat(imperative/functional): let elemwise support empty IO & add some tests
GitOrigin-RevId: a5dc3b997c
|
4 years ago |
Megvii Engine Team
|
809d5056cd
|
feat(mge/distributed): enable pt shm allreduce
GitOrigin-RevId: 1dd5a02a51
|
4 years ago |
Megvii Engine Team
|
0245594156
|
test(autograd): test jvp emulated by 2nd grad
GitOrigin-RevId: 47114fcd99
|
3 years ago |
Megvii Engine Team
|
8480302da8
|
fix(autograd): make higher order grad experimental
GitOrigin-RevId: 81e1eb0ebf
|
4 years ago |
Megvii Engine Team
|
72531f2b4a
|
test(autograd): add more tests for higher order grad
GitOrigin-RevId: 5fc308f87a
|
4 years ago |