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
|
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
|
d4c71f92dc
|
perf(profiler): reduce profiler host overhead
GitOrigin-RevId: 92cea560f5
|
3 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
|
3 years ago |
Megvii Engine Team
|
a829270489
|
fix(mge/tracing): replace detach as fast path copy
GitOrigin-RevId: d765725d5a
|
3 years ago |
Megvii Engine Team
|
32ad926559
|
feat(mge): improve presentation of async errors
GitOrigin-RevId: af5426c37d
|
3 years ago |
Megvii Engine Team
|
9279104b11
|
feat(mge): add opdef serialization and apply_module_trace
GitOrigin-RevId: 5b45bded1d
|
3 years ago |
Megvii Engine Team
|
2a063f8e87
|
fix(subgraph): fix scope mismatch of subgraph content
GitOrigin-RevId: 6e23456250
|
3 years ago |
Megvii Engine Team
|
8c47c1f149
|
perf(syncbn): reimplement with subgraph
GitOrigin-RevId: 13e7e3d3c0
|
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
|
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
|
1e3af4dd17
|
fix(mgb/comp_node): add more info in `comp_node.to_string()`
GitOrigin-RevId: 794a8847aa
|
3 years ago |
Megvii Engine Team
|
b3d2a632f3
|
fix(sublinear): fix lb_memory in sublinear
GitOrigin-RevId: fd51023c01
|
3 years ago |
Megvii Engine Team
|
5f558042b2
|
fix(imperative/ops): use tblgen to generate FastpathCopy
GitOrigin-RevId: b02157dacb
|
3 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
|
3 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 |
Megvii Engine Team
|
b7e596b4a1
|
perf(autograd): copy inputs before capture in backward_graph_grad_rule
GitOrigin-RevId: 8b9c067b2d
|
4 years ago |
Megvii Engine Team
|
2ac3c9dc8b
|
fix(trace): constants in backward graph treat as ImmutableTensor corectlly
GitOrigin-RevId: 5fd6a5e00c
|
4 years ago |
Megvii Engine Team
|
3e4e4c4604
|
feat(mgb/jit): add graph_opt_config and jit_config interfaces
GitOrigin-RevId: 170d9eeab2
|
4 years ago |
Megvii Engine Team
|
355153e158
|
feat(mge/dtr): add DTR in computing graph
GitOrigin-RevId: 8941810319
|
4 years ago |
Megvii Engine Team
|
0ac642b5d5
|
fix(imperative): persistent cache write through on put
GitOrigin-RevId: f9408ae504
|
4 years ago |
Megvii Engine Team
|
86b69cacd0
|
fix(dnn): fixes for int4
GitOrigin-RevId: 845e164fd3
|
4 years ago |
Megvii Engine Team
|
241b35a697
|
refactor(ops): remove BackwardGraph op
GitOrigin-RevId: eda20e5760
|
4 years ago |
Megvii Engine Team
|
8a918717c0
|
feat(mgb): add megengine inference
GitOrigin-RevId: 6ffec6b418
|
4 years ago |
Megvii Engine Team
|
e417798f9e
|
fix(mge): correct pytype when calling apply from python
GitOrigin-RevId: 6abfa06ada
|
4 years ago |
Megvii Engine Team
|
efc7f04fd3
|
refactor(mge/jit): using static global_enable for apply ctx insted of global variable
GitOrigin-RevId: dd82b53faf
|
4 years ago |
Megvii Engine Team
|
750bde9513
|
refactor(mge/jit): remove is_compiled flag in cpp tensor
GitOrigin-RevId: 15f90af735
|
4 years ago |
Megvii Engine Team
|
4f240ec2d3
|
refactor(mge/jit): make trace return any kind of output
GitOrigin-RevId: fd1265c661
|
4 years ago |
Megvii Engine Team
|
717b88e684
|
fix(mge/elemwise): fix problem that elemwise.mode is not comparable with string mode
GitOrigin-RevId: 82e39be0a9
|
4 years ago |
Megvii Engine Team
|
813dd2f1c0
|
fix(mge/core): support non-Tensor args for `core.tensor.utils.get_device`
GitOrigin-RevId: 51dc03db9e
|
4 years ago |
Megvii Engine Team
|
9f5bb42d78
|
perf(fastrun): cache persistent cache
GitOrigin-RevId: 21f7d4c19d
|
4 years ago |
Megvii Engine Team
|
7af1ec66db
|
fix(tensor): del valid tensors when compnode finalizing
GitOrigin-RevId: bace1f2b51
|
4 years ago |