Megvii Engine Team
|
d56570d929
|
fix(megbrain): add rdnn to copybara
GitOrigin-RevId: 7d8bf77053
|
3 years ago |
Megvii Engine Team
|
7de1bb11ab
|
fix(mge/utils): disable memory forwarding for subgraph
GitOrigin-RevId: 3be6d5fe6b
|
3 years ago |
Megvii Engine Team
|
b7c9361f81
|
perf(mge/functional): add infer_output_attrs_fallible for some ops
GitOrigin-RevId: 33ae4b18e9
|
3 years ago |
Megvii Engine Team
|
a4327c4d25
|
perf(imperative): add dim_expansion transform for conv/bn1d
GitOrigin-RevId: d14a69424d
|
3 years ago |
Megvii Engine Team
|
72a70dd6a7
|
perf(imperative): specialize convolution implementation
GitOrigin-RevId: 33634c550f
|
3 years ago |
Megvii Engine Team
|
0a6f4a880e
|
fix(mge/dtr): fix dtr problem
GitOrigin-RevId: 2a703f9ee4
|
3 years ago |
Megvii Engine Team
|
529b394f9c
|
fix(imperative): fix profiler problem
GitOrigin-RevId: ea0d8a5054
|
3 years ago |
Megvii Engine Team
|
e64536a31e
|
fix(imperative): fix the dtype promote problem when amp
GitOrigin-RevId: 43e1035fc8
|
3 years ago |
Megvii Engine Team
|
2b80806f21
|
perf(imperative/src): improve dot performance
GitOrigin-RevId: 35b5bd164f
|
3 years ago |
Megvii Engine Team
|
1709b3940b
|
perf(mge/functional): speed up Broadcast and Reshape
GitOrigin-RevId: a72f5460b6
|
3 years ago |
Megvii Engine Team
|
3e206d899b
|
perf(mge/functional): speed up Split
GitOrigin-RevId: 43550a0706
|
3 years ago |
Megvii Engine Team
|
730ddc2d81
|
perf(interpreter): improve interpreter performance
GitOrigin-RevId: 88f51d15f8
|
3 years ago |
Megvii Engine Team
|
729242f9f8
|
refactor(imperative): move typecvt code of sereval ops to c++
GitOrigin-RevId: 4ffaa376c1
|
3 years ago |
Megvii Engine Team
|
3c3fc6f33c
|
refactor(imperative): move python code of elemwise/reduce/conv2d/bn to c++
GitOrigin-RevId: 01b5324392
|
3 years ago |
Megvii Engine Team
|
8446626193
|
perf(imperative/src): improve elemwise
GitOrigin-RevId: 78aa487277
|
3 years ago |
Megvii Engine Team
|
e400b7ffe5
|
perf(imperative): enable memory forwarding for imperative
GitOrigin-RevId: 7c1993979c
|
4 years ago |
Megvii Engine Team
|
84d1a440f0
|
fix(imperative): do not use output_desc in rng ops
GitOrigin-RevId: e6a399be17
|
3 years ago |
Megvii Engine Team
|
0cb60d646d
|
feat(imperative): add output_descs for apply_on_physical_tensor
GitOrigin-RevId: 5b036c2c5a
|
3 years ago |
Megvii Engine Team
|
c7ded2fe2f
|
refactor(imperative): remove unnecessary reverve in small vector
GitOrigin-RevId: 85c30bc828
|
3 years ago |
Megvii Engine Team
|
8c2b916ef5
|
refactor(imperative): remove some methods in proxy graph
GitOrigin-RevId: 1fb68a1da2
|
3 years ago |
Megvii Engine Team
|
2348a963f2
|
refactor(imperative): apply workspace limit hook to mini graph
GitOrigin-RevId: 27c51f3147
|
3 years ago |
Megvii Engine Team
|
fea46ea9a4
|
perf(imperative): add opr cache for apply_on_physical_tensor
GitOrigin-RevId: fc5d5fb34d
|
4 years ago |
Megvii Engine Team
|
177001d5e5
|
refactor(dispatch): allow dynamic type creation
GitOrigin-RevId: 27dde05cff
|
3 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
|
7bd848ce04
|
fix(subgraph): fix hand-written backward for serval jit-elemwise ops
GitOrigin-RevId: ea3a40d96e
|
4 years ago |
Megvii Engine Team
|
7be7656c9f
|
fix(imperative): explicitly manage global structures
GitOrigin-RevId: 0f910c34b6
|
4 years ago |
Megvii Engine Team
|
62034fb262
|
fix(imperative): make CompNode finalize happens before global object destructor
GitOrigin-RevId: 9a1f507c69
|
4 years ago |
Megvii Engine Team
|
b6ce02a152
|
fix(subgraph): fallback back to cg if jit unsupported
GitOrigin-RevId: 853a00a402
|
4 years ago |
Megvii Engine Team
|
225045236b
|
perf(imperative): improve shape inference
GitOrigin-RevId: 98b4d7e9af
|
4 years ago |
Megvii Engine Team
|
2775f4580c
|
feat(subgraph): subgraph builder supports jit and custom grad
GitOrigin-RevId: e1a1ebdf1c
|
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
|
1c1e9b002d
|
fix(rng): init layout strides
GitOrigin-RevId: 9833d866da
|
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
|
cbb47089a6
|
perf(interpreter): add fastpath for GetVarShape
GitOrigin-RevId: d1ac4e7fe3
|
4 years ago |
Megvii Engine Team
|
b458178847
|
feat(opr): add mutable tensor opr
GitOrigin-RevId: 7f8a3d7b66
|
4 years ago |
Megvii Engine Team
|
93ceb80ad2
|
refactor(imperative): fix broadcast,reshape,reduce
GitOrigin-RevId: ee3dc1487d
|
3 years ago |
Megvii Engine Team
|
e6706be23a
|
refactor(imperative): remove infer_output_mem_desc
GitOrigin-RevId: bff62b33a0
|
4 years ago |
Megvii Engine Team
|
a5af35c18c
|
refactor(imperative): remove command buffer
GitOrigin-RevId: 83c8cb6d3b
|
4 years ago |
Megvii Engine Team
|
c34a75d0f4
|
fix(trace): assume result is not scalar when shape is valid
GitOrigin-RevId: beee2d0f28
|
4 years ago |
Megvii Engine Team
|
95ac055538
|
feat(dnn,mgb,imperative): add diag opr implement
GitOrigin-RevId: 43016ffa2b
|
4 years ago |
Megvii Engine Team
|
cbbca5fb10
|
feat(mge): add softmax op use cudnn api
GitOrigin-RevId: 7734ebf8c4
|
4 years ago |
Megvii Engine Team
|
2faa6ea5a9
|
Merge pull request #213 from kxz18:rnn
GitOrigin-RevId: 9e9215c115
|
4 years ago |
Megvii Engine Team
|
f5b8fec4ca
|
fix(imperative): remove big tensor from host side
GitOrigin-RevId: 2047982d73
|
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 |