Megvii Engine Team
|
2065f0c26c
|
fix(mge/subgraph): fix mem leak
GitOrigin-RevId: dc7bfe6265
|
4 years ago |
Megvii Engine Team
|
57e197475b
|
fix(imperative): check async error when getting value
GitOrigin-RevId: 3945a9bfa2
|
4 years ago |
Megvii Engine Team
|
f64a2e02c9
|
Revert "fix(mge/subgraph): fix mem leak"
This reverts commit 48beb625e1.
GitOrigin-RevId: 5fc58d5903
|
4 years ago |
Megvii Engine Team
|
ffbfe59c9f
|
fix(mge/subgraph): fix mem leak
GitOrigin-RevId: 48beb625e1
|
4 years ago |
Megvii Engine Team
|
0a56a5b110
|
feat(cuda/comp_node): gets (maximum) GPU memory allocated/reserved
GitOrigin-RevId: da2cc22436
|
4 years ago |
Megvii Engine Team
|
a2a09ef990
|
fix(imperative): release dtr related resources when disable dtr
GitOrigin-RevId: eacfded9de
|
4 years ago |
Megvii Engine Team
|
2676fb7344
|
refactor(imperative): remove enable_defrag interface as it is enabled by default
GitOrigin-RevId: de9e7d7f16
|
4 years ago |
Megvii Engine Team
|
d909950f64
|
refactor(imperative): remove swap in dtr
GitOrigin-RevId: 5c9e42f74a
|
4 years ago |
Megvii Engine Team
|
a404c508e9
|
feat(mge): support dump with specific format
GitOrigin-RevId: 57a7c0de02
|
4 years ago |
Megvii Engine Team
|
182ca25d45
|
feat(mge/imperative): add coalesce_all_free_memory mge api
GitOrigin-RevId: ea05b1e9a1
|
4 years ago |
Megvii Engine Team
|
676b205bd8
|
style(imperative): add bypass of clang-format for cg options
GitOrigin-RevId: 824d610631
|
4 years ago |
Megvii Engine Team
|
fe15239ac0
|
fix(imperative): fix error message for tensors with intbx data type
GitOrigin-RevId: cbb42f8127
|
4 years ago |
Megvii Engine Team
|
369c2ccc5a
|
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
|
4 years ago |
Megvii Engine Team
|
000517c641
|
fix(grad): stop using exception in grad_override
GitOrigin-RevId: 00ae38d48b
|
4 years ago |
Megvii Engine Team
|
b6142bee9a
|
feat(imperative): support tensor with uint16 date type
GitOrigin-RevId: 57ba0633c7
|
4 years ago |
Megvii Engine Team
|
65d554ed8e
|
refactor(dtr): improve dtr in computing graph
GitOrigin-RevId: d599a08e52
|
4 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
|
4 years ago |
Megvii Engine Team
|
b3e958d0bc
|
fix(src): fix the warnings and copy.bara.sky in custom op
GitOrigin-RevId: 4ade455897
|
4 years ago |
Megvii Engine Team
|
90dd07161c
|
refactor(imperative): modify the python interface of custom op
GitOrigin-RevId: e82e5de480
|
4 years ago |
Megvii Engine Team
|
cbf024bf20
|
feat(imperative): add adaptor between custom op and imperative runtime
GitOrigin-RevId: d7877f2e32
|
4 years ago |
Megvii Engine Team
|
d4c71f92dc
|
perf(profiler): reduce profiler host overhead
GitOrigin-RevId: 92cea560f5
|
4 years ago |
Megvii Engine Team
|
a841d8e76f
|
refactor(profiler): refactor event processing
GitOrigin-RevId: 26f0c5a6e1
|
4 years ago |
Megvii Engine Team
|
0a665ea452
|
feat(mge/device): enable to get cuda compute capability
GitOrigin-RevId: b5d3f2225c
|
4 years ago |
Megvii Engine Team
|
a829270489
|
fix(mge/tracing): replace detach as fast path copy
GitOrigin-RevId: d765725d5a
|
4 years ago |
Megvii Engine Team
|
32ad926559
|
feat(mge): improve presentation of async errors
GitOrigin-RevId: af5426c37d
|
4 years ago |
Megvii Engine Team
|
9279104b11
|
feat(mge): add opdef serialization and apply_module_trace
GitOrigin-RevId: 5b45bded1d
|
4 years ago |
Megvii Engine Team
|
2a063f8e87
|
fix(subgraph): fix scope mismatch of subgraph content
GitOrigin-RevId: 6e23456250
|
4 years ago |
Megvii Engine Team
|
8c47c1f149
|
perf(syncbn): reimplement with subgraph
GitOrigin-RevId: 13e7e3d3c0
|
4 years ago |
Megvii Engine Team
|
48db45d123
|
perf(interpreter): try put device value with host to reduce d2h
GitOrigin-RevId: 63d36e7706
|
4 years ago |
Megvii Engine Team
|
a605f38b26
|
refactor(opmeth): add OpMethCache struct
GitOrigin-RevId: c1ebe15672
|
4 years ago |
Megvii Engine Team
|
0b8dc2c98b
|
refactor(subgraph): add generic encoded_graph
GitOrigin-RevId: 56d90be0e7
|
4 years ago |
Megvii Engine Team
|
88b3c84229
|
refactor(subgraph): move to subgraph.h
GitOrigin-RevId: 2791f335d4
|
4 years ago |
Megvii Engine Team
|
1e3af4dd17
|
fix(mgb/comp_node): add more info in `comp_node.to_string()`
GitOrigin-RevId: 794a8847aa
|
4 years ago |
Megvii Engine Team
|
b3d2a632f3
|
fix(sublinear): fix lb_memory in sublinear
GitOrigin-RevId: fd51023c01
|
4 years ago |
Megvii Engine Team
|
5f558042b2
|
fix(imperative/ops): use tblgen to generate FastpathCopy
GitOrigin-RevId: b02157dacb
|
4 years ago |
Megvii Engine Team
|
f12355f727
|
fix(imperative/grad): fix hardcode dtype in subtensor_grad_rule
GitOrigin-RevId: 50da4af26d
|
4 years ago |
Megvii Engine Team
|
ffe2bb2eb2
|
fix(mge): fix some errors caused by unknown shape when using symbolic trace or building graph
GitOrigin-RevId: 70ddc06eee
|
4 years ago |
Megvii Engine Team
|
2d42455fa8
|
fix(mge/utils): fix toposort to get definition order
GitOrigin-RevId: 47a26dd6dd
|
4 years ago |
Megvii Engine Team
|
2cd9823210
|
fix(mgb/tensorrt): fix trt runtime, padding channel to a multiple of 4 when using kCHW4 IOFormat
GitOrigin-RevId: c5f1ed70da
|
4 years ago |
Megvii Engine Team
|
b078dda90b
|
feat(mge/random): add some random op and remove random/distrbution.py
GitOrigin-RevId: 4c05ebc266
|
4 years ago |
Megvii Engine Team
|
eb66681f45
|
fix(mge/random): fix delete_rng_handel
GitOrigin-RevId: a836e13c23
|
4 years ago |
Megvii Engine Team
|
0245594156
|
test(autograd): test jvp emulated by 2nd grad
GitOrigin-RevId: 47114fcd99
|
4 years ago |
Megvii Engine Team
|
8480302da8
|
fix(autograd): make higher order grad experimental
GitOrigin-RevId: 81e1eb0ebf
|
4 years ago |
Megvii Engine Team
|
72531f2b4a
|
test(autograd): add more tests for higher order grad
GitOrigin-RevId: 5fc308f87a
|
4 years ago |
Megvii Engine Team
|
522e556b83
|
feat(autodiff): support higher order grad
GitOrigin-RevId: 86390d2179
|
4 years ago |
Megvii Engine Team
|
aa4e847692
|
fix(interpreter): release gil when interpreter blocking
GitOrigin-RevId: c48e9efa5b
|
4 years ago |
Megvii Engine Team
|
9d47c3babd
|
feat(profiler): imperative profiler support tracing
GitOrigin-RevId: b247472feb
|
4 years ago |
Megvii Engine Team
|
1d64792b41
|
refactor(profiler): detach profiler from interpreter
GitOrigin-RevId: f3954728d1
|
4 years ago |
Megvii Engine Team
|
54a4d70eb5
|
feat(src/serialization): add support of serializing metadata
GitOrigin-RevId: b563c94451
|
4 years ago |