Megvii Engine Team
|
6cefabe734
|
fix(dnn/cuda): fix ci
GitOrigin-RevId: 8267e5f9dd
|
4 years ago |
Megvii Engine Team
|
888f4e46ae
|
feat(dnn/cuda): add implicit bmm large kernel dwconv2d dgrad kernels
GitOrigin-RevId: fcb7974d62
|
4 years ago |
Megvii Engine Team
|
08d8635ff5
|
feat(dnn/cuda): add implicit bmm large kernel dwconv2d fprop impl
GitOrigin-RevId: feb09ebb58
|
4 years ago |
Megvii Engine Team
|
93ceb80ad2
|
refactor(imperative): fix broadcast,reshape,reduce
GitOrigin-RevId: ee3dc1487d
|
3 years ago |
Megvii Engine Team
|
d919aaebc7
|
test(imperative): reopen special interpolate test and sync when test rng
GitOrigin-RevId: e3d03b4d1d
|
4 years ago |
Megvii Engine Team
|
ca2deebc0f
|
fix(imperative/tensor): make @ operator has the same functionality as matmul functional
GitOrigin-RevId: bf6136cc1a
|
4 years ago |
Megvii Engine Team
|
e860a08386
|
refactor(mge/indexing): move indexing into c++
GitOrigin-RevId: 43fbdb22dd
|
4 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
|
bdb853ee6f
|
fix(mgb): fix extra device malloc when load MultipleDeviceTensorWithFormatHolder
GitOrigin-RevId: adf4a7f77a
|
4 years ago |
Megvii Engine Team
|
406115dba0
|
fix(imperative): syncbn fp16 support
GitOrigin-RevId: 6059d5b76b
|
4 years ago |
Megvii Engine Team
|
d5ef792309
|
perf(lite): optimized lite tensor get data by share
GitOrigin-RevId: 62e48ca539
|
4 years ago |
huangxinda
|
ce9ad07a27
|
feat(ci): update ci and readme
|
4 years ago |
Megvii Engine Team
|
884865703d
|
test(trace): test subtensor on unknown shape
GitOrigin-RevId: 1b5cfa4e0a
|
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
|
bebb2cf4c3
|
Merge pull request #428 from P2Oileen:fix-pad
GitOrigin-RevId: f33ea46ad6
|
4 years ago |
Megvii Engine Team
|
e2b79ea00e
|
feat(mgb): reduce the number of trtruntimeopr create contexts
GitOrigin-RevId: 14e5d1769e
|
4 years ago |
Megvii Engine Team
|
6157d9cfef
|
fix(traced_module): fix Module compatible issue and traced module getattr check
GitOrigin-RevId: 62eb3bfb10
|
4 years ago |
Megvii Engine Team
|
26b52a61de
|
feat(lite): add get model infomation before create network interface
GitOrigin-RevId: e499f3ebf8
|
4 years ago |
Megvii Engine Team
|
5e17b3e4c6
|
Merge pull request #426 from Qsingle:fix-pixel_suffle
GitOrigin-RevId: db9a0f7551
|
4 years ago |
Megvii Engine Team
|
2bebe80e93
|
fix(imperative): fix the default pickle protocol version of save
GitOrigin-RevId: fab2dc7369
|
4 years ago |
Xinran Xu
|
f02cd2d28b
|
Merge pull request #436 from bealwang/master
docs(readme): add more badges
|
4 years ago |
王彪
|
df4153dc71
|
docs(readme): add more badges
|
4 years ago |
XindaH
|
ea91babbce
|
Merge pull request #435 from MegEngine/try-import
|
4 years ago |
Megvii Engine Team
|
8e94af9d78
|
Merge pull request #400 from jieli-matrix:docstring-svd
GitOrigin-RevId: 3bcbea3440
|
4 years ago |
Megvii Engine Team
|
260923e11c
|
perf(aarch64): optimize aarch64 uint16 relayout with block_w==3
GitOrigin-RevId: fe6aaaac0c
|
4 years ago |
Megvii Engine Team
|
b04c3d1456
|
feat(lite): add set address ptr pair interface
GitOrigin-RevId: 285dacb4da
|
4 years ago |
Megvii Engine Team
|
17f2dffb5b
|
fix(imperative/cpu/multithread): fix multithread at imperative
GitOrigin-RevId: 9120d5cb48
|
4 years ago |
Megvii Engine Team
|
3159eecadd
|
fix(init): fix fan_in and fan_out for group conv2d
GitOrigin-RevId: a6f41063f0
|
4 years ago |
XindaH
|
51c03f3ebd
|
Merge pull request #434 from MegEngine/try-import
|
4 years ago |
zhanghaolong
|
3a27ee0134
|
Update README.md
|
4 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
|
4 years ago |
XindaH
|
8548cc7e54
|
Merge pull request #389 from weixiao-huang/docs/auto-correct
|
4 years ago |
weixiao-huang
|
6d63520691
|
docs(mge): add space between Chinese and English for README_CN.md
|
4 years ago |
Megvii Engine Team
|
275b63114d
|
fix(imperative): fix use collections error from python3.10
GitOrigin-RevId: 5dd019b336
|
4 years ago |
Megvii Engine Team
|
95ac055538
|
feat(dnn,mgb,imperative): add diag opr implement
GitOrigin-RevId: 43016ffa2b
|
4 years ago |
Megvii Engine Team
|
39d77fb55a
|
feat(arm): add arm rnn_cell/lstm_cell/lstm optimized kernel
GitOrigin-RevId: b9bb7352bc
|
4 years ago |
Megvii Engine Team
|
3ddc32d3e3
|
feat(android/whl): support android whl
GitOrigin-RevId: 05df16b494
|
4 years ago |
Megvii Engine Team
|
f509b1be9b
|
fix(build): split elemwise_multi_type cpp
GitOrigin-RevId: 13267e9db6
|
4 years ago |
Megvii Engine Team
|
3252016e05
|
Merge pull request #401 from LosReturn:patch-1
GitOrigin-RevId: 440af8bd3d
|
4 years ago |
Megvii Engine Team
|
f7e034b506
|
feat(lite): add global layout transform python interface for lite
GitOrigin-RevId: f159f49208
|
4 years ago |
Megvii Engine Team
|
e70c07a223
|
feat(lite): add global layout transform c/c++ interface for lite
GitOrigin-RevId: 36a4b26b42
|
4 years ago |
Megvii Engine Team
|
86ee4638bf
|
Merge pull request #402 from AA1HSHH:docstring-reshape
GitOrigin-RevId: 1ec572eb7c
|
4 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
|
4 years ago |
Megvii Engine Team
|
2c2df83051
|
fix(cmake): enable custom op when building develop to avoid the pytest fail
GitOrigin-RevId: fa05ead899
|
4 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
|
4 years ago |
Megvii Engine Team
|
7ea104d788
|
Revert "fix(mge): replace _full_sync by sync"
This reverts commit e36ef45464
GitOrigin-RevId: 2d913c8ac9
|
4 years ago |
Megvii Engine Team
|
cbbca5fb10
|
feat(mge): add softmax op use cudnn api
GitOrigin-RevId: 7734ebf8c4
|
4 years ago |
Megvii Engine Team
|
1d2510b6d7
|
fix(module): fix module dumped in old version without _short_name attr
GitOrigin-RevId: a1c815f613
|
4 years ago |
Megvii Engine Team
|
cf5e9488bb
|
fix(traced_module): fix module trace transformation
GitOrigin-RevId: ce11fe5e09
|
4 years ago |