Megvii Engine Team
|
df3474ca1d
|
perf(functional): rewrite serval elemwise ops with jit subgraph
GitOrigin-RevId: 26247e21d9
|
3 years ago |
Megvii Engine Team
|
c55fda9a7c
|
fix(fastrun): don't kill profiling worker
GitOrigin-RevId: 99a0f11a5a
|
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
|
47fe766310
|
feat(dnn/cuda): add implicit bmm kernels for large kernel depthwise convolution backward filter opr
GitOrigin-RevId: 932e7689e8
|
3 years ago |
Megvii Engine Team
|
dcc9693582
|
feat(dnn/cuda): add heuristic rule for implicit batched gemm large kernel dwconv2d kernels
GitOrigin-RevId: 2d2c213bfd
|
3 years ago |
Megvii Engine Team
|
6cefabe734
|
fix(dnn/cuda): fix ci
GitOrigin-RevId: 8267e5f9dd
|
3 years ago |
Megvii Engine Team
|
888f4e46ae
|
feat(dnn/cuda): add implicit bmm large kernel dwconv2d dgrad kernels
GitOrigin-RevId: fcb7974d62
|
3 years ago |
Megvii Engine Team
|
08d8635ff5
|
feat(dnn/cuda): add implicit bmm large kernel dwconv2d fprop impl
GitOrigin-RevId: feb09ebb58
|
3 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
|
3 years ago |
Megvii Engine Team
|
ca2deebc0f
|
fix(imperative/tensor): make @ operator has the same functionality as matmul functional
GitOrigin-RevId: bf6136cc1a
|
3 years ago |
Megvii Engine Team
|
e860a08386
|
refactor(mge/indexing): move indexing into c++
GitOrigin-RevId: 43fbdb22dd
|
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
|
bdb853ee6f
|
fix(mgb): fix extra device malloc when load MultipleDeviceTensorWithFormatHolder
GitOrigin-RevId: adf4a7f77a
|
3 years ago |
Megvii Engine Team
|
406115dba0
|
fix(imperative): syncbn fp16 support
GitOrigin-RevId: 6059d5b76b
|
3 years ago |
Megvii Engine Team
|
d5ef792309
|
perf(lite): optimized lite tensor get data by share
GitOrigin-RevId: 62e48ca539
|
3 years ago |
huangxinda
|
ce9ad07a27
|
feat(ci): update ci and readme
|
3 years ago |
Megvii Engine Team
|
884865703d
|
test(trace): test subtensor on unknown shape
GitOrigin-RevId: 1b5cfa4e0a
|
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
|
bebb2cf4c3
|
Merge pull request #428 from P2Oileen:fix-pad
GitOrigin-RevId: f33ea46ad6
|
3 years ago |
Megvii Engine Team
|
e2b79ea00e
|
feat(mgb): reduce the number of trtruntimeopr create contexts
GitOrigin-RevId: 14e5d1769e
|
3 years ago |
Megvii Engine Team
|
6157d9cfef
|
fix(traced_module): fix Module compatible issue and traced module getattr check
GitOrigin-RevId: 62eb3bfb10
|
3 years ago |
Megvii Engine Team
|
26b52a61de
|
feat(lite): add get model infomation before create network interface
GitOrigin-RevId: e499f3ebf8
|
3 years ago |
Megvii Engine Team
|
5e17b3e4c6
|
Merge pull request #426 from Qsingle:fix-pixel_suffle
GitOrigin-RevId: db9a0f7551
|
3 years ago |
Megvii Engine Team
|
2bebe80e93
|
fix(imperative): fix the default pickle protocol version of save
GitOrigin-RevId: fab2dc7369
|
3 years ago |
Xinran Xu
|
f02cd2d28b
|
Merge pull request #436 from bealwang/master
docs(readme): add more badges
|
3 years ago |
王彪
|
df4153dc71
|
docs(readme): add more badges
|
3 years ago |
XindaH
|
ea91babbce
|
Merge pull request #435 from MegEngine/try-import
|
3 years ago |
Megvii Engine Team
|
8e94af9d78
|
Merge pull request #400 from jieli-matrix:docstring-svd
GitOrigin-RevId: 3bcbea3440
|
3 years ago |
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 |