Megvii Engine Team
|
260923e11c
|
perf(aarch64): optimize aarch64 uint16 relayout with block_w==3
GitOrigin-RevId: fe6aaaac0c
|
3 years ago |
Megvii Engine Team
|
b04c3d1456
|
feat(lite): add set address ptr pair interface
GitOrigin-RevId: 285dacb4da
|
3 years ago |
Megvii Engine Team
|
17f2dffb5b
|
fix(imperative/cpu/multithread): fix multithread at imperative
GitOrigin-RevId: 9120d5cb48
|
3 years ago |
Megvii Engine Team
|
3159eecadd
|
fix(init): fix fan_in and fan_out for group conv2d
GitOrigin-RevId: a6f41063f0
|
3 years ago |
XindaH
|
51c03f3ebd
|
Merge pull request #434 from MegEngine/try-import
|
3 years ago |
zhanghaolong
|
3a27ee0134
|
Update README.md
|
3 years ago |
iotboy
|
9809871e13
|
docs(readme): update README.md
From GITHUB 393
COPYBARA_INTEGRATE_REVIEW=https://github.com/MegEngine/MegEngine/pull/393 from mlbo:patch-1 fe10cf5bc7
Author: iotboy <iotboy@163.com>
Closes #393
GitOrigin-RevId: 2dfc762e03
|
3 years ago |
XindaH
|
8548cc7e54
|
Merge pull request #389 from weixiao-huang/docs/auto-correct
|
3 years ago |
weixiao-huang
|
6d63520691
|
docs(mge): add space between Chinese and English for README_CN.md
|
3 years ago |
Megvii Engine Team
|
275b63114d
|
fix(imperative): fix use collections error from python3.10
GitOrigin-RevId: 5dd019b336
|
3 years ago |
Megvii Engine Team
|
95ac055538
|
feat(dnn,mgb,imperative): add diag opr implement
GitOrigin-RevId: 43016ffa2b
|
3 years ago |
Megvii Engine Team
|
39d77fb55a
|
feat(arm): add arm rnn_cell/lstm_cell/lstm optimized kernel
GitOrigin-RevId: b9bb7352bc
|
3 years ago |
Megvii Engine Team
|
3ddc32d3e3
|
feat(android/whl): support android whl
GitOrigin-RevId: 05df16b494
|
3 years ago |
Megvii Engine Team
|
f509b1be9b
|
fix(build): split elemwise_multi_type cpp
GitOrigin-RevId: 13267e9db6
|
3 years ago |
Megvii Engine Team
|
3252016e05
|
Merge pull request #401 from LosReturn:patch-1
GitOrigin-RevId: 440af8bd3d
|
3 years ago |
Megvii Engine Team
|
f7e034b506
|
feat(lite): add global layout transform python interface for lite
GitOrigin-RevId: f159f49208
|
3 years ago |
Megvii Engine Team
|
e70c07a223
|
feat(lite): add global layout transform c/c++ interface for lite
GitOrigin-RevId: 36a4b26b42
|
3 years ago |
Megvii Engine Team
|
86ee4638bf
|
Merge pull request #402 from AA1HSHH:docstring-reshape
GitOrigin-RevId: 1ec572eb7c
|
3 years ago |
Megvii Engine Team
|
3251f50114
|
fix(mgb/cuda-stub): add libcuda-wrap_11.4.h to fit the CUDA11.4 toolchain
GitOrigin-RevId: efa38f00d1
|
3 years ago |
Megvii Engine Team
|
2c2df83051
|
fix(cmake): enable custom op when building develop to avoid the pytest fail
GitOrigin-RevId: fa05ead899
|
3 years ago |
Megvii Engine Team
|
ee0b95e935
|
feat(dnn/elemwise/arm_common): support part of arm ternary elemwise multithread
BCAST111C_VEC_BCAST111C and BCAST101_VEC_BCAST101
GitOrigin-RevId: 0e26553c90
|
3 years ago |
Megvii Engine Team
|
7ea104d788
|
Revert "fix(mge): replace _full_sync by sync"
This reverts commit e36ef45464
GitOrigin-RevId: 2d913c8ac9
|
3 years ago |
Megvii Engine Team
|
cbbca5fb10
|
feat(mge): add softmax op use cudnn api
GitOrigin-RevId: 7734ebf8c4
|
3 years ago |
Megvii Engine Team
|
1d2510b6d7
|
fix(module): fix module dumped in old version without _short_name attr
GitOrigin-RevId: a1c815f613
|
3 years ago |
Megvii Engine Team
|
cf5e9488bb
|
fix(traced_module): fix module trace transformation
GitOrigin-RevId: ce11fe5e09
|
3 years ago |
Megvii Engine Team
|
97c90d9137
|
feat(traced_module): add _exclude_from_trace
GitOrigin-RevId: 615b769a02
|
3 years ago |
Megvii Engine Team
|
30e565e5b8
|
fix(traced_module): fix error message
GitOrigin-RevId: 3046225e30
|
3 years ago |
Megvii Engine Team
|
de8ffe0c12
|
refactor(imperative): unify interpreter option setting
GitOrigin-RevId: 53510445cc
|
3 years ago |
Megvii Engine Team
|
8b60bdfa10
|
fix(mge): replace _full_sync by sync
GitOrigin-RevId: e36ef45464
|
3 years ago |
Megvii Engine Team
|
20b42a8c3b
|
fix(dnn): add naive lstm kernel
GitOrigin-RevId: f08ef810cf
|
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
|
68cde8734e
|
fix(mge/imperative): support broadcast with None
GitOrigin-RevId: dd330a2a1d
|
3 years ago |
Megvii Engine Team
|
0bdd0b1467
|
refactor(dispatch): switch to new dispatch system
GitOrigin-RevId: 32dd49a23a
|
3 years ago |
Megvii Engine Team
|
d3689c3f3c
|
feat(imperative/python): add transformation manager
GitOrigin-RevId: a3c1732ffd
|
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
|
42759dc7f5
|
test(dtr): make dtr_resnet1202 isolated
GitOrigin-RevId: 4d863cc32e
|
3 years ago |
Megvii Engine Team
|
7f03ae9aed
|
fix(imperative): reduce tls usage
GitOrigin-RevId: a716b2ae98
|
3 years ago |
Megvii Engine Team
|
85ea882cb5
|
fix(mgb/ops): immutable tensor support empty storage
GitOrigin-RevId: 2851498fce
|
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
|
c3d63f14a5
|
feat(dispatch): add gdb debug utils
GitOrigin-RevId: 393b0c9c76
|
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 |