Megvii Engine Team
194201a774
revert: chore(mge/misc): api converage
This reverts commit --
77fd432cb2
0d7126641b
559d205a8c
6d8e7398e4
b72517bfe9
GitOrigin-RevId: 18dc4ce999
2 years ago
Megvii Engine Team
98460f583f
chore(mge/misc): rename APIs that should not be public
GitOrigin-RevId: b72517bfe9
2 years ago
Megvii Engine Team
d2a1905ad5
Revert "feat(mgb): add cumprod opr"
This reverts commit 3436c3bdaa
.
GitOrigin-RevId: 95ab3d1aa7
2 years ago
Megvii Engine Team
49e14f87b5
feat(mgb): add cumprod opr
GitOrigin-RevId: 3436c3bdaa
3 years ago
Megvii Engine Team
a49d5bf973
fix(autodiff): fix inplace operation on autodiff.Function
GitOrigin-RevId: a658680f35
3 years ago
Megvii Engine Team
2484cd2741
fix(tensor): check args when construct tensor with existing tensor
GitOrigin-RevId: 0345454070
3 years ago
Megvii Engine Team
24c5c19bf0
fix(imperative): make functional ops support negative axis
GitOrigin-RevId: f61e01270b
3 years ago
Megvii Engine Team
e694301721
fix(ops/jit): skip lookup include path when nvcc executable not found
GitOrigin-RevId: f5e7dce1c5
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
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
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
e393d1cf65
feat(mge/amp): add convert_format module for NHWC training
GitOrigin-RevId: 1b41e1042c
3 years ago
Megvii Engine Team
533fb5bf49
feat(imperative): support formatted tensor and add special op rules
GitOrigin-RevId: 77ff909f23
3 years ago
Megvii Engine Team
4aa79c453b
perf(mge): override grad of matmul
GitOrigin-RevId: d9d97e70fe
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
787a22a9d6
perf(tensor): implement __new__ in cpp
GitOrigin-RevId: 4defd249c3
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
877bda4180
perf(mge): improve cross stream memory borrowing
GitOrigin-RevId: c68977c5dc
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
2f3bc2db9d
perf(mge/utils): move astensor1d into C++
GitOrigin-RevId: e7c6659020
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
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
884865703d
test(trace): test subtensor on unknown shape
GitOrigin-RevId: 1b5cfa4e0a
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
de8ffe0c12
refactor(imperative): unify interpreter option setting
GitOrigin-RevId: 53510445cc
3 years ago
Megvii Engine Team
0bdd0b1467
refactor(dispatch): switch to new dispatch system
GitOrigin-RevId: 32dd49a23a
3 years ago
Megvii Engine Team
d24f198cfa
fix(imperative): clear custom op build cache
GitOrigin-RevId: cb145fe1ad
3 years ago
Megvii Engine Team
619d78ed86
fix(imperative): check async error when getting value
GitOrigin-RevId: 52b8a29932
3 years ago
Megvii Engine Team
a72e0cb568
feat(imperative,src): add jit builder for custom op
GitOrigin-RevId: 3bb0b46311
3 years ago
Megvii Engine Team
7b19bc76fb
feat(traced_module): support traced module backward compatible serialization
GitOrigin-RevId: aaa9e51c74
3 years ago
Megvii Engine Team
68cdabd288
feat(opr): indexing_multi_axis_vec support nd index
GitOrigin-RevId: 07b1248bdc
3 years ago
Megvii Engine Team
0ddabb06cc
fix(imperative/astype): fix astype when target dtype is int4
GitOrigin-RevId: 1c5195176d
3 years ago
Megvii Engine Team
d909950f64
refactor(imperative): remove swap in dtr
GitOrigin-RevId: 5c9e42f74a
3 years ago
Megvii Engine Team
b6142bee9a
feat(imperative): support tensor with uint16 date type
GitOrigin-RevId: 57ba0633c7
3 years ago
Megvii Engine Team
dc28a43409
fix(mgb/bn): fix empty tensor input problem and other minor problems
GitOrigin-RevId: 1708b76cb8
3 years ago
Megvii Engine Team
32ad926559
feat(mge): improve presentation of async errors
GitOrigin-RevId: af5426c37d
3 years ago