Megvii Engine Team
df5ebd3da7
fix(imperative/ops): fix the vmemory problem in 1.9
GitOrigin-RevId: 728e983701
3 years ago
Megvii Engine Team
da91e650a5
refactor(ops/layer_norm): speed up the host speed of layer_norm
GitOrigin-RevId: 6f359b5b29
3 years ago
Megvii Engine Team
261a5bce23
feat(imperative/amp): add dimshuffle in set_format for nhwc
GitOrigin-RevId: 5ced9e1a31
3 years ago
Megvii Engine Team
3892aa0b6e
fix(imperative/amp): fix bn params for nhwc amp
GitOrigin-RevId: 57a3b9d418
3 years ago
Megvii Engine Team
c28a875fac
fix(imperative/amp): adapt new transformation
GitOrigin-RevId: 6edd577a70
3 years ago
Megvii Engine Team
fd41302cc1
feat(imperative/amp): add set_format
GitOrigin-RevId: 91de6f49de
3 years ago
Megvii Engine Team
fc633ce4ff
fix(imperative/amp): fix custom grad in Subgraph
GitOrigin-RevId: 1c728d6ab9
3 years ago
Megvii Engine Team
673b295d75
feat(imperative/amp): remove conv_format and bn param_dim configs
GitOrigin-RevId: 848d34f63d
3 years ago
Megvii Engine Team
e393d1cf65
feat(mge/amp): add convert_format module for NHWC training
GitOrigin-RevId: 1b41e1042c
3 years ago
Megvii Engine Team
a926878c01
feat(imperative): remove symbolvar of imperative
GitOrigin-RevId: 16da6d1491
3 years ago
Megvii Engine Team
c2e9860feb
chore(license): remove all license in file header
GitOrigin-RevId: a0e31247a6
3 years ago
Megvii Engine Team
8182af6eb6
fix(mgb): fix strategy of grad_op and opr_attr
GitOrigin-RevId: bb7ab8fa9d
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
c2435d1561
perf(imperative): specialize adaptive pooling
GitOrigin-RevId: 01e1418458
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
2d72de8a87
fix(mge): fix infer output attrs fallible
GitOrigin-RevId: ea18c7f753
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
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
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
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
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
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
Megvii Engine Team
a93741815b
feat(mgb/opr): add layernorm forward and backward kernel
GitOrigin-RevId: 0cd484e753
3 years ago
P2Oileen
5d755898fd
fix(funtional): correct the typo: witdth to width
3 years ago
Qsingle
14499e83d9
fix the type error of pixel shuffle
add the test of dtype float
3 years ago
Megvii Engine Team
7cb3ad8a3e
fix(mge/functional): fix one_hot irregular coding style
From GITHUB 399
ORIGINAL_AUTHOR=Asthestarsfalll <1186454801@qq.com>
COPYBARA_INTEGRATE_REVIEW=https://github.com/MegEngine/MegEngine/pull/399 from Asthestarsfalll:master 541bf3af29
GITHUB_PUBLIC_PR_NUMBER=399
GITHUB_PR_URL=https://github.com/MegEngine/MegEngine/pull/399
GitOrigin-RevId: 5df007207a
3 years ago
Megvii Engine Team
7ac693a882
docs(imperative): fix multiple target warning when cross-reference
GitOrigin-RevId: c5b4884c8a
3 years ago
Megvii Engine Team
ea3eb66e7f
docs(mge): modify some docstring about conv and tools
GitOrigin-RevId: 4833e78d71
3 years ago
Megvii Engine Team
5c94db16f2
feat(mge/functional): add groups support for conv_transpose2d & 3d
GitOrigin-RevId: b75b792fb4
3 years ago
Megvii Engine Team
8a222d2c9c
fix(mge): replace functional paramter
GitOrigin-RevId: af3c68c588
3 years ago
Megvii Engine Team
29f9935d92
fix(imperative/python): add layer_norm doc and rm useless param
GitOrigin-RevId: 1b15db621e
3 years ago
Megvii Engine Team
2108696588
feat(mge/imperative): add local response normalization
GitOrigin-RevId: 939a4d26dd
3 years ago
Megvii Engine Team
ca4c93dee7
fix(mge): fix layer norm amp bug
GitOrigin-RevId: dba691fcbf
3 years ago