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
|
3 years ago |
Megvii Engine Team
|
81d8c73a41
|
perf(dispatch/trace): serval tricks to speed up trace
GitOrigin-RevId: 2bdd70cde2
|
3 years ago |
Megvii Engine Team
|
4fa6162027
|
perf(dispatch): improve performance of dispatch system
GitOrigin-RevId: 860028e1af
|
3 years ago |
Megvii Engine Team
|
ca00177719
|
perf(dispatch): speed up dispatch system
GitOrigin-RevId: eabbe3e021
|
3 years ago |
Megvii Engine Team
|
7bd848ce04
|
fix(subgraph): fix hand-written backward for serval jit-elemwise ops
GitOrigin-RevId: ea3a40d96e
|
3 years ago |
Megvii Engine Team
|
7be7656c9f
|
fix(imperative): explicitly manage global structures
GitOrigin-RevId: 0f910c34b6
|
3 years ago |
Megvii Engine Team
|
62034fb262
|
fix(imperative): make CompNode finalize happens before global object destructor
GitOrigin-RevId: 9a1f507c69
|
3 years ago |
Megvii Engine Team
|
b6ce02a152
|
fix(subgraph): fallback back to cg if jit unsupported
GitOrigin-RevId: 853a00a402
|
3 years ago |
Megvii Engine Team
|
225045236b
|
perf(imperative): improve shape inference
GitOrigin-RevId: 98b4d7e9af
|
3 years ago |
Megvii Engine Team
|
2775f4580c
|
feat(subgraph): subgraph builder supports jit and custom grad
GitOrigin-RevId: e1a1ebdf1c
|
3 years ago |
Megvii Engine Team
|
3c61e0e02a
|
feat(ops): add JITFusion op
GitOrigin-RevId: 7dc35d4e80
|
3 years ago |
Megvii Engine Team
|
aa587446fc
|
feat(subgraph): support shape inference for CompiledOp
GitOrigin-RevId: a96b8f3446
|
3 years ago |
Megvii Engine Team
|
1c1e9b002d
|
fix(rng): init layout strides
GitOrigin-RevId: 9833d866da
|
3 years ago |
Megvii Engine Team
|
9527859cc8
|
feat(opcache): add ndim and has_value to cache key
GitOrigin-RevId: ad073d389e
|
3 years ago |
Megvii Engine Team
|
cbb47089a6
|
perf(interpreter): add fastpath for GetVarShape
GitOrigin-RevId: d1ac4e7fe3
|
3 years ago |
Megvii Engine Team
|
b458178847
|
feat(opr): add mutable tensor opr
GitOrigin-RevId: 7f8a3d7b66
|
3 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
|
3 years ago |
Megvii Engine Team
|
a5af35c18c
|
refactor(imperative): remove command buffer
GitOrigin-RevId: 83c8cb6d3b
|
3 years ago |
Megvii Engine Team
|
c34a75d0f4
|
fix(trace): assume result is not scalar when shape is valid
GitOrigin-RevId: beee2d0f28
|
3 years ago |
Megvii Engine Team
|
95ac055538
|
feat(dnn,mgb,imperative): add diag opr implement
GitOrigin-RevId: 43016ffa2b
|
3 years ago |
Megvii Engine Team
|
cbbca5fb10
|
feat(mge): add softmax op use cudnn api
GitOrigin-RevId: 7734ebf8c4
|
3 years ago |
Megvii Engine Team
|
2faa6ea5a9
|
Merge pull request #213 from kxz18:rnn
GitOrigin-RevId: 9e9215c115
|
3 years ago |
Megvii Engine Team
|
f5b8fec4ca
|
fix(imperative): remove big tensor from host side
GitOrigin-RevId: 2047982d73
|
3 years ago |
Megvii Engine Team
|
9ce1f0f5d1
|
refactor(dispatch): implement grad
GitOrigin-RevId: d8367f9587
|
3 years ago |
Megvii Engine Team
|
c609c031f1
|
refactor(dispatch): implement symbol
GitOrigin-RevId: c7bd86f5c1
|
3 years ago |
Megvii Engine Team
|
e32929dfd2
|
refactor(dispatch): implement scalar
GitOrigin-RevId: b244c2ca1a
|
3 years ago |
Megvii Engine Team
|
59084fa857
|
refactor(dispatch): implement lazy_eval
GitOrigin-RevId: 4e3f3a1c44
|
3 years ago |
Megvii Engine Team
|
d2b67c2a88
|
refactor(dispatch): implement trace
GitOrigin-RevId: f8d3005732
|
3 years ago |
Megvii Engine Team
|
39ac606b9c
|
refactor(dispatch): implement eval
GitOrigin-RevId: 32563e0a27
|
3 years ago |
Megvii Engine Team
|
7f03ae9aed
|
fix(imperative): reduce tls usage
GitOrigin-RevId: a716b2ae98
|
3 years ago |
Megvii Engine Team
|
d8d5edb31f
|
refactor(opdef): overwrite shared_from_this to avoid const_cast
GitOrigin-RevId: 5322dbbb3f
|
3 years ago |
Megvii Engine Team
|
4b0ecb5deb
|
fix(ops/recv): use std::vector to store shape to support scalar
GitOrigin-RevId: e1dac3c919
|
3 years ago |
Megvii Engine Team
|
2e42bc083d
|
feat(dispatch): implement new op dispatch system
GitOrigin-RevId: 355da6b814
|
3 years ago |
Megvii Engine Team
|
2be6ceda03
|
feat(imperative/utils): add serval utils
GitOrigin-RevId: f401663ae3
|
3 years ago |
Megvii Engine Team
|
e7c2ed1183
|
fix(imperative): add missing pragma once
GitOrigin-RevId: 7a9fff1171
|
3 years ago |
Megvii Engine Team
|
462364eb06
|
perf(interpreter): don't check host value if unnecessary
GitOrigin-RevId: 5306c71328
|
3 years ago |
Megvii Engine Team
|
619d78ed86
|
fix(imperative): check async error when getting value
GitOrigin-RevId: 52b8a29932
|
3 years ago |
Megvii Engine Team
|
1999307015
|
feat(mgb/opr): add dropout kernel
GitOrigin-RevId: d248bd2005
|
3 years ago |
Megvii Engine Team
|
a93741815b
|
feat(mgb/opr): add layernorm forward and backward kernel
GitOrigin-RevId: 0cd484e753
|
3 years ago |
Megvii Engine Team
|
1657b8e881
|
fix(fastrun): fix persistent_cache in redis
GitOrigin-RevId: ada5862b05
|
3 years ago |
Megvii Engine Team
|
6011f51001
|
style(all): fix clang-format for MGB_DEFINE inside another macro
GitOrigin-RevId: 8c2b6a2aed
|
3 years ago |
Megvii Engine Team
|
6b28a2f9ec
|
fix(copybara): re-open redis cache
GitOrigin-RevId: 055bf6aa4f
|
3 years ago |