Megvii Engine Team
|
eca6e1d931
|
fix(ci): fixes for ci
GitOrigin-RevId: b0a432bd2e
|
3 years ago |
Megvii Engine Team
|
19d7412a0a
|
refactor(mgb/gopt): reorganize code of global layout transform
GitOrigin-RevId: 4973820e02
|
3 years ago |
Megvii Engine Team
|
8ef12bdf22
|
feat(mgb/gopt): add user inferface for global layout transform
GitOrigin-RevId: b71d6c60ef
|
3 years ago |
Megvii Engine Team
|
a3cd3fc74f
|
test(mgb/gopt): add testcase for global layout transform
GitOrigin-RevId: f9669e1ba0
|
3 years ago |
Megvii Engine Team
|
af576e9ae0
|
fix(mgb/gopt): fix auto padding for nhwc layout
GitOrigin-RevId: 038e372cbe
|
3 years ago |
Megvii Engine Team
|
af828ca9ec
|
feat(mgb/gopt): fix folding conv dimshuffle pass
GitOrigin-RevId: 756878c173
|
3 years ago |
Megvii Engine Team
|
c67c4b7ddc
|
feat(mgb/gopt): add layout transform pass
GitOrigin-RevId: 9ed5f5782e
|
3 years ago |
Megvii Engine Team
|
2ec7c16725
|
feat(mgb/gopt): profiler support opr filter and var node filter
GitOrigin-RevId: 5f8d86687f
|
3 years ago |
Megvii Engine Team
|
50ea5ae8e2
|
feat(mgb/gopt): add dynamic programming solver
GitOrigin-RevId: 595392ec89
|
3 years ago |
Megvii Engine Team
|
c14e5719f8
|
feat(mgb/gopt): add profile impl for global layout transform pass
GitOrigin-RevId: 8ef62baf79
|
3 years ago |
Megvii Engine Team
|
9c0a17d07b
|
feat(mgb/gopt): add auto aligned reformat impls
GitOrigin-RevId: fd0814fdb3
|
3 years ago |
Megvii Engine Team
|
2ed76b1660
|
feat(mgb/gopt): add graph dumper for graph partition
GitOrigin-RevId: 6dbcb67009
|
3 years ago |
Megvii Engine Team
|
76b2840858
|
feat(mgb/gopt): add subgraph extractor
GitOrigin-RevId: 56fd701c2c
|
3 years ago |
Megvii Engine Team
|
8a3eb05a1b
|
refactor(mgb/gopt): refactor tensor reformat opt pass
GitOrigin-RevId: a1b1e89b76
|
3 years ago |
Megvii Engine Team
|
c33126ab5c
|
feat(mgb/gopt): add reformat manager
GitOrigin-RevId: b9791b131a
|
3 years ago |
Megvii Engine Team
|
65d554ed8e
|
refactor(dtr): improve dtr in computing graph
GitOrigin-RevId: d599a08e52
|
3 years ago |
Megvii Engine Team
|
4f28e14684
|
fix(dnn): fix compatibility broken of convolution format
GitOrigin-RevId: c17f02d03d
|
3 years ago |
Megvii Engine Team
|
f642b05e17
|
test(mge): update traced_module unit test
GitOrigin-RevId: 3948d50d79
|
3 years ago |
Megvii Engine Team
|
fb20cb36a0
|
docs(mge/traced_module): update traced_module api doc
GitOrigin-RevId: 19a95d26c7
|
3 years ago |
Megvii Engine Team
|
c7a8d945c7
|
fix(mge/traced_module): let graph record total_id
GitOrigin-RevId: f99178f3ac
|
3 years ago |
Megvii Engine Team
|
8b40f57738
|
feat(mgb/dnn): add conv1x1 algo for matrix mul
GitOrigin-RevId: 585b2c045a
|
3 years ago |
Megvii Engine Team
|
4de62ad6cd
|
docs(mge/device): fix docstring of get_cuda_compute_capability
GitOrigin-RevId: 4ec04e8b9b
|
3 years ago |
Megvii Engine Team
|
fb49a2834f
|
refactor(mgb/dnn): refactor enum used in serializing
GitOrigin-RevId: e57af4a59c
|
3 years ago |
Megvii Engine Team
|
d69b59035d
|
feat(dnn): add an get_all_algorithms_safe interface
GitOrigin-RevId: e3734e4531
|
3 years ago |
Megvii Engine Team
|
103d7f33ba
|
refactor(dnn/rocm): update hip license header
GitOrigin-RevId: 79d684755d
|
4 years ago |
Megvii Engine Team
|
5aa52d3863
|
feat(dnn/rocm): add adaptive pooling opr
GitOrigin-RevId: e844b3e770
|
3 years ago |
Megvii Engine Team
|
83cf4ee64e
|
refactor(dnn/rocm): remove some useless includes
GitOrigin-RevId: 3d2c315a36
|
3 years ago |
Megvii Engine Team
|
323a4642e6
|
feat(dnn/rocm): add topk opr
GitOrigin-RevId: 5ecb079854
|
3 years ago |
Megvii Engine Team
|
f4784f4af1
|
feat(dnn/rocm): add argsort opr
GitOrigin-RevId: b4c3eb4707
|
3 years ago |
Megvii Engine Team
|
6082c353e7
|
feat(dnn/rocm): support bool in type_cvt and elemwise
GitOrigin-RevId: ad5ec7bc1c
|
4 years ago |
Megvii Engine Team
|
0ad377c7cf
|
fix(mgb/gopt): add error message when input dtype is not equal to param dtype in BN2Elemwise pass
GitOrigin-RevId: 3d09a2a12e
|
3 years ago |
Megvii Engine Team
|
93f70a9539
|
feat(src): add MGB_CUSTOM_OP macro
GitOrigin-RevId: beddcf9745
|
3 years ago |
Megvii Engine Team
|
b3e958d0bc
|
fix(src): fix the warnings and copy.bara.sky in custom op
GitOrigin-RevId: 4ade455897
|
3 years ago |
Megvii Engine Team
|
cdb692d2fa
|
refactor(imperative): add TODO tag for some functions
GitOrigin-RevId: e295a1fa55
|
3 years ago |
Megvii Engine Team
|
90dd07161c
|
refactor(imperative): modify the python interface of custom op
GitOrigin-RevId: e82e5de480
|
3 years ago |
Megvii Engine Team
|
cbf024bf20
|
feat(imperative): add adaptor between custom op and imperative runtime
GitOrigin-RevId: d7877f2e32
|
3 years ago |
Megvii Engine Team
|
39ba30214d
|
feat(src): add adaptor between custom op and graph runtime
GitOrigin-RevId: 834c0ac72b
|
3 years ago |
Megvii Engine Team
|
c9134717c6
|
feat(src/custom): add tensor and tensor attrs in custom namespace
GitOrigin-RevId: 9f69259205
|
3 years ago |
Megvii Engine Team
|
43fb2c4afb
|
feat(opr): let roll support empty IO
GitOrigin-RevId: b9a59b623a
|
3 years ago |
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
|
3bc94738ba
|
fix(profiler): do not assume clock duration and time_point type
GitOrigin-RevId: 2015bc30bd
|
3 years ago |
Megvii Engine Team
|
b7e6bd7bba
|
feat(profiler): custom event support device
GitOrigin-RevId: 4709c44b39
|
3 years ago |
Megvii Engine Team
|
defbc20efb
|
fix(profiler): fix profiler in dtr
GitOrigin-RevId: 79b5a6b526
|
3 years ago |
Megvii Engine Team
|
d4c71f92dc
|
perf(profiler): reduce profiler host overhead
GitOrigin-RevId: 92cea560f5
|
3 years ago |
Megvii Engine Team
|
8fddd8089d
|
fix(profiler): respect record_device option
GitOrigin-RevId: 7c9a8cfba7
|
4 years ago |