Megvii Engine Team
25f97b76ea
feat(imperative): load_nerwork_and_run enable weight preprocess
GitOrigin-RevId: 0642b237a7
2 years ago
Megvii Engine Team
bccda5c427
fix(mgb/imperative): fix repeat bug in trace mode
GitOrigin-RevId: 9547fc6102
2 years ago
Megvii Engine Team
fac67e7c2b
feat(gopt): support nchw44 global pooling with fuse_grain
GitOrigin-RevId: 4c43a149f8
2 years ago
Megvii Engine Team
d60d028a40
feat(mge/device): enable to get cuda/cudnn/tensorrt version
GitOrigin-RevId: 5864c61d10
3 years ago
Megvii Engine Team
51397bfc08
feat(mgb): supports value infer and empty input tensor in ElemwiseMultiType
GitOrigin-RevId: 05577a8bc8
3 years ago
Megvii Engine Team
ac51f7807d
feat(mge/distributed): add support for batch send recv op
GitOrigin-RevId: eb3d712704
3 years ago
Megvii Engine Team
9d3977279e
fix(traced_module): fix bug of renaming tesnor
GitOrigin-RevId: 468a996bdd
3 years ago
Megvii Engine Team
7252825ce4
fix(functional): broadcast_to supports mutable target shape
GitOrigin-RevId: ff79456d5d
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
1192a9a61b
fix(imperative): fix adaptive pool2d
GitOrigin-RevId: 624b47a51a
3 years ago
Megvii Engine Team
7b4b94fd93
fix(imperative): fix the segmentfault when reduce backward
GitOrigin-RevId: 8a3e63d4f5
3 years ago
Megvii Engine Team
24c5c19bf0
fix(imperative): make functional ops support negative axis
GitOrigin-RevId: f61e01270b
3 years ago
Megvii Engine Team
c76e80bc8e
fix(mge): fix mem leak of numpy method of DeviceTensorND
GitOrigin-RevId: 06082145f6
3 years ago
Megvii Engine Team
eef0308b29
feat(serialization): add no_change_graph and version param whem dump model
GitOrigin-RevId: 65064452c9
3 years ago
Megvii Engine Team
b9a6932341
feat(imperative): channel default model format to fbs v2
GitOrigin-RevId: 6066516c31
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
cd26376549
style(imperative/amp): reformat code
GitOrigin-RevId: 6e5a6e1eaf
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
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
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
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
1783b8977a
feat(profiler): integrate cupti backend
GitOrigin-RevId: dec8be1908
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
597a1e791b
refactor(imperative): add interface to clear algorithm cache
GitOrigin-RevId: 662618954b
3 years ago
Megvii Engine Team
b92866d2c2
fix(build): fix build depends dirty file issue
GitOrigin-RevId: 435d8b5c50
3 years ago
Megvii Engine Team
27d4c4b36c
refactor(stats): use static inline variable declaration
GitOrigin-RevId: 7d86e5f257
3 years ago
Megvii Engine Team
787a22a9d6
perf(tensor): implement __new__ in cpp
GitOrigin-RevId: 4defd249c3
3 years ago
Megvii Engine Team
7005b7d691
fix(mge): fix fastpath check
GitOrigin-RevId: acc8957469
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
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
1c2a323e78
feat(mge): add warning message when mismatched cuda sm is detected
GitOrigin-RevId: f78c79eb06
3 years ago
Megvii Engine Team
877bda4180
perf(mge): improve cross stream memory borrowing
GitOrigin-RevId: c68977c5dc
3 years ago
Megvii Engine Team
c2435d1561
perf(imperative): specialize adaptive pooling
GitOrigin-RevId: 01e1418458
3 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
2a900a69cb
perf(imperative): improve reduce op performance
GitOrigin-RevId: 26d982a7b8
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
6c413ba943
refactor(mge): refactor physical tensor
GitOrigin-RevId: 93ba67ca5a
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
e64536a31e
fix(imperative): fix the dtype promote problem when amp
GitOrigin-RevId: 43e1035fc8
3 years ago
Megvii Engine Team
2f3bc2db9d
perf(mge/utils): move astensor1d into C++
GitOrigin-RevId: e7c6659020
3 years ago