Megvii Engine Team
67cfce9f2f
fix(imperative/amp): add is_scalar check in elemwise and concat
GitOrigin-RevId: 61a612e92a
3 years ago
Megvii Engine Team
d313f92610
fix(imperative/amp): fix format transformation for symbol trans
GitOrigin-RevId: 96cc237c67
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
c9e56f4987
feat(imperative/amp): add dimshuffle before creating nhwc tensor
GitOrigin-RevId: 4461f9a0d3
3 years ago
Megvii Engine Team
d57a071271
feat(imperative/amp): add fallback for op not supported for nhwc tensor
GitOrigin-RevId: 8411ce7bdc
3 years ago
Megvii Engine Team
38a9aa9faf
feat(imperative/amp): add auto dimshuffle for elemwise and concat
GitOrigin-RevId: 6e3df4e064
3 years ago
Megvii Engine Team
cd26376549
style(imperative/amp): reformat code
GitOrigin-RevId: 6e5a6e1eaf
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
7e9aa742e6
feat(imperative/amp): enable auto_convert_format by default
GitOrigin-RevId: 71ae311fed
3 years ago
Megvii Engine Team
fc0f454685
fix(dnn/check_non_finite): adjust some details of CheckNonFinite
GitOrigin-RevId: 52ddd805b4
3 years ago
Megvii Engine Team
e393d1cf65
feat(mge/amp): add convert_format module for NHWC training
GitOrigin-RevId: 1b41e1042c
4 years ago
Megvii Engine Team
533fb5bf49
feat(imperative): support formatted tensor and add special op rules
GitOrigin-RevId: 77ff909f23
4 years ago
Megvii Engine Team
4aa79c453b
perf(mge): override grad of matmul
GitOrigin-RevId: d9d97e70fe
3 years ago
Megvii Engine Team
98b5ee78c1
feat(mge/dnn): add lamb optimizer
GitOrigin-RevId: 5a27157456
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
58df717e37
fix(mge/autodiff): fix attaching tensor already in gradient path
GitOrigin-RevId: da774509ca
3 years ago
Megvii Engine Team
597a1e791b
refactor(imperative): add interface to clear algorithm cache
GitOrigin-RevId: 662618954b
3 years ago
Megvii Engine Team
787a22a9d6
perf(tensor): implement __new__ in cpp
GitOrigin-RevId: 4defd249c3
3 years ago
Megvii Engine Team
7bf5b0ee1e
test(imperative): check env values after each pytest
GitOrigin-RevId: 826788113a
3 years ago
Megvii Engine Team
811579b0e7
fix(imperative): restrict value converts to symbolvar
GitOrigin-RevId: 271267be69
3 years ago
Megvii Engine Team
99a85c4079
fix(mge): fix advanced indexing grad
GitOrigin-RevId: 8033c9322d
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
877bda4180
perf(mge): improve cross stream memory borrowing
GitOrigin-RevId: c68977c5dc
4 years ago
Megvii Engine Team
90cedd3c1d
fix(imperative): restrict using convert_inputs in py_apply
GitOrigin-RevId: b021aac8a6
3 years ago
Megvii Engine Team
fd6f8e58b0
feat(mgb/dtype): add dtype qint1
GitOrigin-RevId: abe9fb68b1
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
bc9aa47ad8
feat(mge/indexing): support newaxis
GitOrigin-RevId: 8338c4b475
3 years ago
Megvii Engine Team
9779bc7f6d
fix(imperative): allow rng op infer shape fallible
GitOrigin-RevId: 687844500c
3 years ago
Megvii Engine Team
2f3bc2db9d
perf(mge/utils): move astensor1d into C++
GitOrigin-RevId: e7c6659020
3 years ago
Megvii Engine Team
1709b3940b
perf(mge/functional): speed up Broadcast and Reshape
GitOrigin-RevId: a72f5460b6
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
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
87f00232f2
fix(mge/gm): fix missing dtype checking while attach tensors
GitOrigin-RevId: f0aaea99b9
3 years ago
Megvii Engine Team
ab6d12caff
feat(mge): add conv padding mode
GitOrigin-RevId: 147ced856e
4 years ago
Megvii Engine Team
27346b0b65
test(opr): add scalar check for opr_test
GitOrigin-RevId: dcfd7ad5d6
4 years ago
Megvii Engine Team
df3474ca1d
perf(functional): rewrite serval elemwise ops with jit subgraph
GitOrigin-RevId: 26247e21d9
4 years ago
Megvii Engine Team
d919aaebc7
test(imperative): reopen special interpolate test and sync when test rng
GitOrigin-RevId: e3d03b4d1d
4 years ago
Megvii Engine Team
ca2deebc0f
fix(imperative/tensor): make @ operator has the same functionality as matmul functional
GitOrigin-RevId: bf6136cc1a
4 years ago
Megvii Engine Team
e860a08386
refactor(mge/indexing): move indexing into c++
GitOrigin-RevId: 43fbdb22dd
4 years ago
Megvii Engine Team
e6706be23a
refactor(imperative): remove infer_output_mem_desc
GitOrigin-RevId: bff62b33a0
4 years ago
Megvii Engine Team
a5af35c18c
refactor(imperative): remove command buffer
GitOrigin-RevId: 83c8cb6d3b
4 years ago
Megvii Engine Team
406115dba0
fix(imperative): syncbn fp16 support
GitOrigin-RevId: 6059d5b76b
4 years ago
Megvii Engine Team
884865703d
test(trace): test subtensor on unknown shape
GitOrigin-RevId: 1b5cfa4e0a
4 years ago