Megvii Engine Team
2ed38f457f
feat(test): add some comment and test in imperative
GitOrigin-RevId: c2363abe29
3 years ago
Megvii Engine Team
b8c7557b86
fix(mm): fix mm error when use sync
GitOrigin-RevId: 63387bda04
3 years ago
Megvii Engine Team
c49d3070ba
refactor(imperative/ops): extends DnnOprCaller with template
GitOrigin-RevId: 402cba209a
4 years ago
Megvii Engine Team
d19fc2c1da
fix(imperative): add alloc TensorPtr in imperative
GitOrigin-RevId: 1b438fc436
4 years ago
Megvii Engine Team
ac51f7807d
feat(mge/distributed): add support for batch send recv op
GitOrigin-RevId: eb3d712704
4 years ago
Megvii Engine Team
2484cd2741
fix(tensor): check args when construct tensor with existing tensor
GitOrigin-RevId: 0345454070
4 years ago
Megvii Engine Team
02c1a0c3de
fix(imperative/format): warn once when parameter without format been attached
GitOrigin-RevId: 02174447f6
4 years ago
Megvii Engine Team
d313f92610
fix(imperative/amp): fix format transformation for symbol trans
GitOrigin-RevId: 96cc237c67
4 years ago
Megvii Engine Team
cd26376549
style(imperative/amp): reformat code
GitOrigin-RevId: 6e5a6e1eaf
4 years ago
Megvii Engine Team
6f0b582064
chore(imperative/amp): adapt dev
GitOrigin-RevId: 41eb0faadf
4 years ago
Megvii Engine Team
c28a875fac
fix(imperative/amp): adapt new transformation
GitOrigin-RevId: 6edd577a70
4 years ago
Megvii Engine Team
fd41302cc1
feat(imperative/amp): add set_format
GitOrigin-RevId: 91de6f49de
4 years ago
Megvii Engine Team
fc633ce4ff
fix(imperative/amp): fix custom grad in Subgraph
GitOrigin-RevId: 1c728d6ab9
4 years ago
Megvii Engine Team
7e9aa742e6
feat(imperative/amp): enable auto_convert_format by default
GitOrigin-RevId: 71ae311fed
4 years ago
Megvii Engine Team
533fb5bf49
feat(imperative): support formatted tensor and add special op rules
GitOrigin-RevId: 77ff909f23
4 years ago
Megvii Engine Team
a926878c01
feat(imperative): remove symbolvar of imperative
GitOrigin-RevId: 16da6d1491
4 years ago
Megvii Engine Team
c2e9860feb
chore(license): remove all license in file header
GitOrigin-RevId: a0e31247a6
4 years ago
Megvii Engine Team
1783b8977a
feat(profiler): integrate cupti backend
GitOrigin-RevId: dec8be1908
4 years ago
Megvii Engine Team
58df717e37
fix(mge/autodiff): fix attaching tensor already in gradient path
GitOrigin-RevId: da774509ca
4 years ago
Megvii Engine Team
8182af6eb6
fix(mgb): fix strategy of grad_op and opr_attr
GitOrigin-RevId: bb7ab8fa9d
4 years ago
Megvii Engine Team
4fb3d886b8
perf(misc): use reinterpret_cast to convert valueshape
GitOrigin-RevId: ee14c7ce5e
4 years ago
Megvii Engine Team
27d4c4b36c
refactor(stats): use static inline variable declaration
GitOrigin-RevId: 7d86e5f257
4 years ago
Megvii Engine Team
877bda4180
perf(mge): improve cross stream memory borrowing
GitOrigin-RevId: c68977c5dc
4 years ago
Megvii Engine Team
95a30eb6f6
perf(imperative): speed up stackmanager guard
GitOrigin-RevId: 12d23b6f7e
4 years ago
Megvii Engine Team
2a900a69cb
perf(imperative): improve reduce op performance
GitOrigin-RevId: 26d982a7b8
4 years ago
Megvii Engine Team
c2293815b2
fix(autodiff): proxy_graph_detail::make_backward_graph support multiple opnodes
GitOrigin-RevId: 2c0c8f330d
4 years ago
Megvii Engine Team
273c0e8745
fix(autodiff): fix some bugs in relation to 2nd order grad
1. implement double backward for batchnorm
2. fix grad attach in nested grad manager
3. pad empty tensor for unsatisfied output_has_grad
4. support double backward for jit subgraph
5. support double backward for autodiff.Function
6. readd debug flag MGE_LOG_OP_DISPATCH
GitOrigin-RevId: cd31ddc620
4 years ago
Megvii Engine Team
a4327c4d25
perf(imperative): add dim_expansion transform for conv/bn1d
GitOrigin-RevId: d14a69424d
4 years ago
Megvii Engine Team
0a6f4a880e
fix(mge/dtr): fix dtr problem
GitOrigin-RevId: 2a703f9ee4
4 years ago
Megvii Engine Team
730ddc2d81
perf(interpreter): improve interpreter performance
GitOrigin-RevId: 88f51d15f8
4 years ago
Megvii Engine Team
3c3fc6f33c
refactor(imperative): move python code of elemwise/reduce/conv2d/bn to c++
GitOrigin-RevId: 01b5324392
4 years ago
Megvii Engine Team
8446626193
perf(imperative/src): improve elemwise
GitOrigin-RevId: 78aa487277
4 years ago
Megvii Engine Team
e400b7ffe5
perf(imperative): enable memory forwarding for imperative
GitOrigin-RevId: 7c1993979c
4 years ago
Megvii Engine Team
0cb60d646d
feat(imperative): add output_descs for apply_on_physical_tensor
GitOrigin-RevId: 5b036c2c5a
4 years ago
Megvii Engine Team
177001d5e5
refactor(dispatch): allow dynamic type creation
GitOrigin-RevId: 27dde05cff
4 years ago
Megvii Engine Team
150a6a6151
perf(dispatch/trace): remove unnecessary h2d for constant
GitOrigin-RevId: d00de3fc1f
4 years ago
Megvii Engine Team
81d8c73a41
perf(dispatch/trace): serval tricks to speed up trace
GitOrigin-RevId: 2bdd70cde2
4 years ago
Megvii Engine Team
4fa6162027
perf(dispatch): improve performance of dispatch system
GitOrigin-RevId: 860028e1af
4 years ago
Megvii Engine Team
ca00177719
perf(dispatch): speed up dispatch system
GitOrigin-RevId: eabbe3e021
4 years ago
Megvii Engine Team
7be7656c9f
fix(imperative): explicitly manage global structures
GitOrigin-RevId: 0f910c34b6
4 years ago
Megvii Engine Team
3c61e0e02a
feat(ops): add JITFusion op
GitOrigin-RevId: 7dc35d4e80
4 years ago
Megvii Engine Team
aa587446fc
feat(subgraph): support shape inference for CompiledOp
GitOrigin-RevId: a96b8f3446
4 years ago
Megvii Engine Team
9527859cc8
feat(opcache): add ndim and has_value to cache key
GitOrigin-RevId: ad073d389e
4 years ago
Megvii Engine Team
b458178847
feat(opr): add mutable tensor opr
GitOrigin-RevId: 7f8a3d7b66
4 years ago
Megvii Engine Team
e6706be23a
refactor(imperative): remove infer_output_mem_desc
GitOrigin-RevId: bff62b33a0
4 years ago
Megvii Engine Team
9ce1f0f5d1
refactor(dispatch): implement grad
GitOrigin-RevId: d8367f9587
4 years ago
Megvii Engine Team
c609c031f1
refactor(dispatch): implement symbol
GitOrigin-RevId: c7bd86f5c1
4 years ago
Megvii Engine Team
e32929dfd2
refactor(dispatch): implement scalar
GitOrigin-RevId: b244c2ca1a
4 years ago
Megvii Engine Team
59084fa857
refactor(dispatch): implement lazy_eval
GitOrigin-RevId: 4e3f3a1c44
4 years ago
Megvii Engine Team
d2b67c2a88
refactor(dispatch): implement trace
GitOrigin-RevId: f8d3005732
4 years ago