Megvii Engine Team
8182af6eb6
fix(mgb): fix strategy of grad_op and opr_attr
GitOrigin-RevId: bb7ab8fa9d
3 years ago
Megvii Engine Team
597a1e791b
refactor(imperative): add interface to clear algorithm cache
GitOrigin-RevId: 662618954b
3 years ago
Megvii Engine Team
dbce6526d6
fix(mge/functional): fix return dtype of comparison function
GitOrigin-RevId: 810e32a829
3 years ago
Megvii Engine Team
811579b0e7
fix(imperative): restrict value converts to symbolvar
GitOrigin-RevId: 271267be69
3 years ago
Megvii Engine Team
409c988163
fix(imperative): add matmul apply_on_varnode
GitOrigin-RevId: 2cf6bf237c
3 years ago
Megvii Engine Team
7927e98fd6
perf(mge): speed up PixelShuffle
GitOrigin-RevId: 942e755745
3 years ago
Megvii Engine Team
f7e10ea8d7
perf(imperative): improve matmul/batch_matmul
GitOrigin-RevId: 4ceb2eb601
3 years ago
Megvii Engine Team
84ce94fbd9
docs(imperative): fix docs about vision related function
GitOrigin-RevId: 55d12d5e7a
3 years ago
Megvii Engine Team
c2435d1561
perf(imperative): specialize adaptive pooling
GitOrigin-RevId: 01e1418458
3 years ago
Megvii Engine Team
8fcbe8258d
docs(mge/functional): fix functional.eye docstring
GitOrigin-RevId: 2645af5e35
3 years ago
Megvii Engine Team
d2278f02d2
perf(imperative): speed up conv_transpose3d
GitOrigin-RevId: e741305446
3 years ago
Megvii Engine Team
3f11b4213a
docs(functional): replace megengine function testcode with doctest format
GitOrigin-RevId: f91d5f5afd
3 years ago
Megvii Engine Team
60163c5d12
docs(functional): replace elem testcode with doctest format
GitOrigin-RevId: a32a011724
3 years ago
Megvii Engine Team
fd5031f8eb
docs(functional): replace utils testcode with doctest format
GitOrigin-RevId: 8618918ff3
3 years ago
Megvii Engine Team
aa63230592
docs(functional): replace loss function testcode with doctest format
GitOrigin-RevId: 98224f0e5f
3 years ago
Megvii Engine Team
2d72de8a87
fix(mge): fix infer output attrs fallible
GitOrigin-RevId: ea18c7f753
3 years ago
Megvii Engine Team
69673f14f7
fix(imperative): remove convert_inputs from concat
GitOrigin-RevId: 1511cb4b43
3 years ago
Megvii Engine Team
22be7e11ca
docs(mge/functional): typo
GitOrigin-RevId: 0ebc6cb322
3 years ago
Megvii Engine Team
111c150db1
docs(mge/functional): fix typo
GitOrigin-RevId: 9473f17970
3 years ago
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
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
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
7bd848ce04
fix(subgraph): fix hand-written backward for serval jit-elemwise ops
GitOrigin-RevId: ea3a40d96e
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
ca2deebc0f
fix(imperative/tensor): make @ operator has the same functionality as matmul functional
GitOrigin-RevId: bf6136cc1a
3 years ago
Megvii Engine Team
406115dba0
fix(imperative): syncbn fp16 support
GitOrigin-RevId: 6059d5b76b
3 years ago
Megvii Engine Team
bebb2cf4c3
Merge pull request #428 from P2Oileen:fix-pad
GitOrigin-RevId: f33ea46ad6
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
8e94af9d78
Merge pull request #400 from jieli-matrix:docstring-svd
GitOrigin-RevId: 3bcbea3440
3 years ago
Megvii Engine Team
95ac055538
feat(dnn,mgb,imperative): add diag opr implement
GitOrigin-RevId: 43016ffa2b
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
cbbca5fb10
feat(mge): add softmax op use cudnn api
GitOrigin-RevId: 7734ebf8c4
3 years ago
Megvii Engine Team
0bdd0b1467
refactor(dispatch): switch to new dispatch system
GitOrigin-RevId: 32dd49a23a
3 years ago
Megvii Engine Team
d23d1352e7
fix(imperative/python): add the default warning for args descending
GitOrigin-RevId: cb5f065e6c
3 years ago
Megvii Engine Team
619d78ed86
fix(imperative): check async error when getting value
GitOrigin-RevId: 52b8a29932
3 years ago
Megvii Engine Team
fba544881b
fix(traced_module): add functional pad to __all__
GitOrigin-RevId: a4fdedfb7f
3 years ago
Megvii Engine Team
1999307015
feat(mgb/opr): add dropout kernel
GitOrigin-RevId: d248bd2005
3 years ago