Megvii Engine Team
273c0e8745
fix(autodiff): fix some bugs in relation to 2nd order grad
1. implement double backward for batchnorm
2. fix grad attach in nested grad manager
3. pad empty tensor for unsatisfied output_has_grad
4. support double backward for jit subgraph
5. support double backward for autodiff.Function
6. readd debug flag MGE_LOG_OP_DISPATCH
GitOrigin-RevId: cd31ddc620
3 years ago
Megvii Engine Team
a4327c4d25
perf(imperative): add dim_expansion transform for conv/bn1d
GitOrigin-RevId: d14a69424d
3 years ago
Megvii Engine Team
2f3bc2db9d
perf(mge/utils): move astensor1d into C++
GitOrigin-RevId: e7c6659020
3 years ago
Megvii Engine Team
fa62f6c06e
perf(mge/utils): move convert_input into C++
GitOrigin-RevId: 0d1cd36251
3 years ago
Megvii Engine Team
d98be08030
perf(mge): move Const into C++
GitOrigin-RevId: 31a443cffd
3 years ago
Megvii Engine Team
1709b3940b
perf(mge/functional): speed up Broadcast and Reshape
GitOrigin-RevId: a72f5460b6
3 years ago
Megvii Engine Team
0f736a0ab4
perf(mge/functional): speed up Dimshuffle
GitOrigin-RevId: 8160c9522b
3 years ago
Megvii Engine Team
3e5e08b0b4
perf(mge/functional): speed up RemoveAxis
GitOrigin-RevId: 9c5d27fe1d
3 years ago
Megvii Engine Team
a4d473c99a
perf(mge/functional): speed up AddAxis
GitOrigin-RevId: 92a3e1bdd3
3 years ago
Megvii Engine Team
3e206d899b
perf(mge/functional): speed up Split
GitOrigin-RevId: 43550a0706
3 years ago
Megvii Engine Team
729242f9f8
refactor(imperative): move typecvt code of sereval ops to c++
GitOrigin-RevId: 4ffaa376c1
3 years ago
Megvii Engine Team
3c3fc6f33c
refactor(imperative): move python code of elemwise/reduce/conv2d/bn to c++
GitOrigin-RevId: 01b5324392
3 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
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
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
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
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
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