Megvii Engine Team
|
34be1d6e57
|
feat(mgb): dump static memory info by tensorboard
GitOrigin-RevId: 2020ad81f9
|
3 years ago |
Megvii Engine Team
|
11a383bc79
|
feat(mgb): add tensorboard tool c++ layer interface and python parsing
GitOrigin-RevId: ae98bc887f
|
3 years ago |
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
|
d69b59035d
|
feat(dnn): add an get_all_algorithms_safe interface
GitOrigin-RevId: e3734e4531
|
3 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
|
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
|
694aa1bd92
|
feat(dnn): add heuristic cache
GitOrigin-RevId: 35e942b5e3
|
3 years ago |
Megvii Engine Team
|
bc9cfc277a
|
feat(mgb): add arm resize nchwxx and naive nearest interp
GitOrigin-RevId: d5fbd59a30
|
3 years ago |
Megvii Engine Team
|
d5cff81c1e
|
fix(mge): limit task queue size
GitOrigin-RevId: a25c3390fc
|
3 years ago |
Megvii Engine Team
|
27f2ecc436
|
feat(imperative/opr): add extern c opr and support dump
GitOrigin-RevId: b29e374c60
|
3 years ago |
Megvii Engine Team
|
43cb305e53
|
feat(mge): add imperative pad
GitOrigin-RevId: de79de536f
|
3 years ago |
Megvii Engine Team
|
567586a037
|
feat(mgb): fastrun algo profile deduplication
GitOrigin-RevId: 0d1bed781d
|
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
|
7a9f2ed95e
|
perf(functional/split): add python binding for Split opr
GitOrigin-RevId: 3359f5da26
|
3 years ago |
Megvii Engine Team
|
ab309eb5fc
|
feat(mgb/opr): let Split support empty IO
GitOrigin-RevId: aad6dc06bf
|
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
|
703b783c24
|
feat(mgb/opr): let Indexing(Set)MultiAxisVec support empty input
GitOrigin-RevId: f15b1d45a1
|
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
|
f9722af3df
|
fix(src/json): fix parsing error when inf is dumped as a Number
GitOrigin-RevId: 62cfba48c0
|
3 years ago |
Megvii Engine Team
|
a4f0e58180
|
fix(mgb/extern_c_opr): throw exception when extern c opr loader was created
GitOrigin-RevId: bf48e1f3bb
|
3 years ago |
Megvii Engine Team
|
36df3850f3
|
test(mgb): remove the padding random test case
padding random test case in ci 2080ti will failed in random
GitOrigin-RevId: b83e7b8799
|
3 years ago |