Megvii Engine Team
b2827cb1b1
feat(opr): let Dot, MatrixMul and BatchedMatrixMul support empty input
GitOrigin-RevId: 10a3c5b106
3 years ago
Megvii Engine Team
50f738772b
feat(opr): let Copy and Identity support empty IO
GitOrigin-RevId: 4e49d8eae8
3 years ago
Megvii Engine Team
74cbc10d82
feat(opr): let batchnorm support empty IO
GitOrigin-RevId: 219411c80c
3 years ago
Megvii Engine Team
8ae61752a0
feat(opr): let nms support empty IO
GitOrigin-RevId: 4c51b1aedb
3 years ago
Megvii Engine Team
1a1748daf1
feat(opr): let Argsort support empty IO
GitOrigin-RevId: 05fcac6e47
3 years ago
Megvii Engine Team
7234efe1e7
feat(opr): let random op support empty output
GitOrigin-RevId: a2174975aa
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
91264f3797
fix(traced_module): fix __getattr__ of TracedModuleBuilder
GitOrigin-RevId: 94d91d6938
3 years ago
Megvii Engine Team
db9ca19633
refactor(traced_module): refactor some testcases and examples
GitOrigin-RevId: e06f733cdd
3 years ago
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
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
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
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
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