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 |
Megvii Engine Team
|
5d9ac970ab
|
fix(mgb): fix fastrun compnode
GitOrigin-RevId: 8db93facb9
|
3 years ago |
Megvii Engine Team
|
6070f1272d
|
fix(mgb): fix getting static memory alloc info
GitOrigin-RevId: dfc69c3b3f
|
4 years ago |
Megvii Engine Team
|
e8a5932d1e
|
perf(mgb/gopt): optimize impl of reformat builders
GitOrigin-RevId: 844b7e8d39
|
3 years ago |
Megvii Engine Team
|
58b8b14554
|
refactor(mgb/gopt): add checker for reformat emitter
GitOrigin-RevId: 53a8c128f5
|
3 years ago |
Megvii Engine Team
|
55efc8e197
|
feat(mgb/gopt): add reformat emitter
GitOrigin-RevId: 937b20a57c
|
4 years ago |
Megvii Engine Team
|
3c6665f7c1
|
feat(lite/whl): merge lite whl to main package
GitOrigin-RevId: 27c7e50207
|
3 years ago |
Megvii Engine Team
|
53da5c79f4
|
feat(cg): add comp_seq_sync_device option
GitOrigin-RevId: c2199c59e9
|
3 years ago |
Megvii Engine Team
|
1e3af4dd17
|
fix(mgb/comp_node): add more info in `comp_node.to_string()`
GitOrigin-RevId: 794a8847aa
|
3 years ago |
Megvii Engine Team
|
4b08e79da8
|
fix(mgb): fix fastrun no_profiling_on_shape_change
GitOrigin-RevId: 2735536424
|
3 years ago |
Megvii Engine Team
|
eab6afab47
|
feat(mgb): add padding opr for megbrain
GitOrigin-RevId: 490e0c5d5a
|
4 years ago |
Megvii Engine Team
|
2224a25205
|
fix(mge/opr): add opr_footprint support for PoolingBackward
GitOrigin-RevId: 5f1c64ef9a
|
3 years ago |
Megvii Engine Team
|
91675a7177
|
fix(mgb): change span-too-large warning to debug level
GitOrigin-RevId: b69e0ba6f3
|
3 years ago |