Megvii Engine Team
|
1ce78aa09b
|
fix(imperative): destruct dnn handles at last
GitOrigin-RevId: 7a67c68c55
|
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
|
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
|
87f00232f2
|
fix(mge/gm): fix missing dtype checking while attach tensors
GitOrigin-RevId: f0aaea99b9
|
3 years ago |
Megvii Engine Team
|
04193e3bd1
|
feat(dnn): add nearest mode for remap and resize
GitOrigin-RevId: 31e7b72a78
|
3 years ago |
Megvii Engine Team
|
69b89388e8
|
docs(mge/functional): fix debug_param set_execution_strategy docstring
GitOrigin-RevId: 434929c998
|
3 years ago |
Megvii Engine Team
|
90c8a58cca
|
docs(docstring): add pad docstring
GitOrigin-RevId: eaf6a87456
|
3 years ago |
Megvii Engine Team
|
ac2f548c9a
|
docs(imperative/dataloader): update preload description
GitOrigin-RevId: 523618b1e1
|
3 years ago |
Megvii Engine Team
|
ab6d12caff
|
feat(mge): add conv padding mode
GitOrigin-RevId: 147ced856e
|
3 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
|
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
|
59cbf9583d
|
fix(subgraph): use CompiledOp in cpu to avoid workspace error
GitOrigin-RevId: 104dd982ef
|
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
|
21f5a7fcc0
|
fix(subgraph): fix device recognition and scalar propagate
GitOrigin-RevId: fd2fe8bec9
|
3 years ago |
Megvii Engine Team
|
27346b0b65
|
test(opr): add scalar check for opr_test
GitOrigin-RevId: dcfd7ad5d6
|
3 years ago |
Megvii Engine Team
|
df3474ca1d
|
perf(functional): rewrite serval elemwise ops with jit subgraph
GitOrigin-RevId: 26247e21d9
|
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
|
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
|
406115dba0
|
fix(imperative): syncbn fp16 support
GitOrigin-RevId: 6059d5b76b
|
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
|
6157d9cfef
|
fix(traced_module): fix Module compatible issue and traced module getattr check
GitOrigin-RevId: 62eb3bfb10
|
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 |
Megvii Engine Team
|
8e94af9d78
|
Merge pull request #400 from jieli-matrix:docstring-svd
GitOrigin-RevId: 3bcbea3440
|
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 |
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
|
3ddc32d3e3
|
feat(android/whl): support android whl
GitOrigin-RevId: 05df16b494
|
3 years ago |
Megvii Engine Team
|
3252016e05
|
Merge pull request #401 from LosReturn:patch-1
GitOrigin-RevId: 440af8bd3d
|
3 years ago |
Megvii Engine Team
|
86ee4638bf
|
Merge pull request #402 from AA1HSHH:docstring-reshape
GitOrigin-RevId: 1ec572eb7c
|
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 |