Megvii Engine Team
|
6ce1483bc9
|
fix(utils/network): fix replace var in different networks
GitOrigin-RevId: 8d1356ddb1
|
3 years ago |
Megvii Engine Team
|
43cb305e53
|
feat(mge): add imperative pad
GitOrigin-RevId: de79de536f
|
3 years ago |
Megvii Engine Team
|
0a665ea452
|
feat(mge/device): enable to get cuda compute capability
GitOrigin-RevId: b5d3f2225c
|
3 years ago |
Megvii Engine Team
|
ab309eb5fc
|
feat(mgb/opr): let Split support empty IO
GitOrigin-RevId: aad6dc06bf
|
3 years ago |
Megvii Engine Team
|
a829270489
|
fix(mge/tracing): replace detach as fast path copy
GitOrigin-RevId: d765725d5a
|
3 years ago |
Megvii Engine Team
|
32ad926559
|
feat(mge): improve presentation of async errors
GitOrigin-RevId: af5426c37d
|
3 years ago |
Megvii Engine Team
|
e6a8b0256e
|
fix(mge): ignore errors caused by earlier async errors
GitOrigin-RevId: ce2028d38a
|
3 years ago |
Megvii Engine Team
|
ee790cb62c
|
fix(imperative/python): fix warp_perspective doc and arange dtype
GitOrigin-RevId: dc4e021ec0
|
3 years ago |
Megvii Engine Team
|
76ce81e828
|
fix(mge): fix F.nn.dropout train and inference bugs
GitOrigin-RevId: 9d9f246d7b
|
3 years ago |
Megvii Engine Team
|
5431929ec7
|
feat(functional): let advance indexing support empty tensor and add more tests
GitOrigin-RevId: 49e1492934
|
3 years ago |
Megvii Engine Team
|
a430c912fa
|
feat(mgb/opr): let CondTake support empty input
GitOrigin-RevId: dfb401a945
|
3 years ago |
Megvii Engine Team
|
432fdb7e6a
|
feat(mgb/opr): let SetSubtensor support empty IO
GitOrigin-RevId: 13909e3b11
|
3 years ago |
Megvii Engine Team
|
e954b8f998
|
feat(mgb/opr): let Subtensor support empty IO
GitOrigin-RevId: a768498104
|
3 years ago |
Megvii Engine Team
|
03e80759b6
|
feat(mge/distributed): add hybird parallel Opr
GitOrigin-RevId: ff26671746
|
3 years ago |
Megvii Engine Team
|
e5217db9f9
|
feat(traced_module): move traced_module out of the experimental folder
GitOrigin-RevId: 36c76b5277
|
3 years ago |
Megvii Engine Team
|
6d1a4f20e7
|
feat(traced_module): support tracing submodules in list/dict
GitOrigin-RevId: 4076b47a89
|
3 years ago |
Megvii Engine Team
|
a3f9073c2c
|
feat(traced_module): update graph transform and add _module_name
GitOrigin-RevId: ef63ae0fd0
|
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
|
442b4f6c26
|
test(traced_module): add some testcases for traced module
GitOrigin-RevId: 0d6bb20b2b
|
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
|
d86ed426ee
|
fix(dtr): simulate the system stack to avoid stack overflow during recomputing
GitOrigin-RevId: cb73e62b19
|
3 years ago |
Megvii Engine Team
|
a7a3bf2d6c
|
test(subgraph): simple test for subgraph
GitOrigin-RevId: 3d6ecd5db7
|
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
|
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
|
0b4a767965
|
feat(mge/distributed): enable uint8 for collective communication
GitOrigin-RevId: 3305c0cf14
|
3 years ago |
Megvii Engine Team
|
3f3a256e0f
|
fix(mge/functional): fix conv* dtype promotion
GitOrigin-RevId: 3f03790cfc
|
3 years ago |
Megvii Engine Team
|
f76a2cc2c6
|
feat(mge/opr): add silu and gelu
GitOrigin-RevId: 75aa42947e
|
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
|
207a346351
|
chore(mge): run get_device_count("gpu") in subprocess
GitOrigin-RevId: 0f0dc001cf
|
4 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
|
4 years ago |
Megvii Engine Team
|
1040b77843
|
fix(mge/functional): fix F.topk(kth_only=True)
GitOrigin-RevId: ddecd1d14b
|
4 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
|
dedecf6922
|
fix(imperative/utils): fix logical error of replace var
GitOrigin-RevId: 614302552c
|
4 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
|
2d42455fa8
|
fix(mge/utils): fix toposort to get definition order
GitOrigin-RevId: 47a26dd6dd
|
4 years ago |