Megvii Engine Team
|
3977b7aa0b
|
feat(mgb/shuffle): add shuffle opr
GitOrigin-RevId: 80490a6f84
|
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
|
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
|
50f738772b
|
feat(opr): let Copy and Identity support empty IO
GitOrigin-RevId: 4e49d8eae8
|
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 |
Megvii Engine Team
|
59d5976622
|
fix(profiler): collect records when thread exit
GitOrigin-RevId: fb796dd61c
|
4 years ago |
Megvii Engine Team
|
a841d8e76f
|
refactor(profiler): refactor event processing
GitOrigin-RevId: 26f0c5a6e1
|
4 years ago |
Megvii Engine Team
|
ca4374b6e2
|
perf(subgraph): remove unnecessary sync
GitOrigin-RevId: 1546286706
|
3 years ago |
Megvii Engine Team
|
ad079009e1
|
fix(interpreter): avoid deadlock in GetValue
GitOrigin-RevId: fd43873167
|
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
|
73d25779ae
|
fix(ops): check index layout for IndexingOneHot
GitOrigin-RevId: 80b42a8de9
|
3 years ago |
Megvii Engine Team
|
43cb305e53
|
feat(mge): add imperative pad
GitOrigin-RevId: de79de536f
|
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
|
e6a8b0256e
|
fix(mge): ignore errors caused by earlier async errors
GitOrigin-RevId: ce2028d38a
|
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
|
d86ed426ee
|
fix(dtr): simulate the system stack to avoid stack overflow during recomputing
GitOrigin-RevId: cb73e62b19
|
3 years ago |
Megvii Engine Team
|
989fdde255
|
refactor(subgraph): use graph queue to cache compiled op graphs
GitOrigin-RevId: cba8574c73
|
3 years ago |
Megvii Engine Team
|
2a063f8e87
|
fix(subgraph): fix scope mismatch of subgraph content
GitOrigin-RevId: 6e23456250
|
3 years ago |
Megvii Engine Team
|
53da5c79f4
|
feat(cg): add comp_seq_sync_device option
GitOrigin-RevId: c2199c59e9
|
3 years ago |
Megvii Engine Team
|
e1c7b22ff0
|
perf(ops): enable memory forward for reduce in special cases
GitOrigin-RevId: dd6e1664c5
|
3 years ago |
Megvii Engine Team
|
cd60d26852
|
perf(ops): specialize Broadcast
GitOrigin-RevId: 0cba3e6e93
|
3 years ago |
Megvii Engine Team
|
3fd3e000d1
|
feat(ops): add serval utility ops
GitOrigin-RevId: 623cb5ddfc
|
3 years ago |
Megvii Engine Team
|
5b4f7c5dd0
|
perf(interpreter): unwind ops with make_forward_graph
GitOrigin-RevId: 5fb8c85089
|
3 years ago |
Megvii Engine Team
|
5798f6ce20
|
feat(subgraph): add OpMeth make_forward_graph
GitOrigin-RevId: 171301fc2b
|
3 years ago |
Megvii Engine Team
|
48db45d123
|
perf(interpreter): try put device value with host to reduce d2h
GitOrigin-RevId: 63d36e7706
|
3 years ago |
Megvii Engine Team
|
a605f38b26
|
refactor(opmeth): add OpMethCache struct
GitOrigin-RevId: c1ebe15672
|
3 years ago |
Megvii Engine Team
|
0213dbe556
|
feat(subgraph): add graph builder
GitOrigin-RevId: f32cfc39e0
|
3 years ago |
Megvii Engine Team
|
0b8dc2c98b
|
refactor(subgraph): add generic encoded_graph
GitOrigin-RevId: 56d90be0e7
|
3 years ago |
Megvii Engine Team
|
88b3c84229
|
refactor(subgraph): move to subgraph.h
GitOrigin-RevId: 2791f335d4
|
3 years ago |
Megvii Engine Team
|
0ef5183c65
|
perf(opdef/reshape): specialize Reshape
GitOrigin-RevId: 26d0e151ca
|
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
|
20e8541bbb
|
refactor(imperative): bind fallback impl on first op method call
GitOrigin-RevId: 82ae1e3205
|
3 years ago |
Megvii Engine Team
|
18274e023c
|
fix(imperative): fix error message when applying custom function with non-tensor arguments
GitOrigin-RevId: 387d6fda4a
|
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
|
dd1fecdf29
|
feat(mge/opr): add cumsum
GitOrigin-RevId: 740f00a8e5
|
3 years ago |
Megvii Engine Team
|
5f558042b2
|
fix(imperative/ops): use tblgen to generate FastpathCopy
GitOrigin-RevId: b02157dacb
|
3 years ago |
Megvii Engine Team
|
7c9569e4e5
|
fix(mge/random): fix random seed
GitOrigin-RevId: 121f459b1b
|
3 years ago |
Megvii Engine Team
|
56b94d89a2
|
feat(dtr): add sqrt sampling
GitOrigin-RevId: 8cb2ceb520
|
3 years ago |