Megvii Engine Team
|
b3f46734e7
|
feat(megdnn/softmax): add softmax operator in fallback
GitOrigin-RevId: 97bc32f561
|
3 years ago |
Megvii Engine Team
|
6c78e68451
|
fix(lite): fix lite memory leak
GitOrigin-RevId: 075c686162
|
2 years ago |
Megvii Engine Team
|
ff239c638a
|
feat(lite): add unit test for lar
GitOrigin-RevId: f3ba1e8a9a
|
2 years ago |
Megvii Engine Team
|
7bf1c38c74
|
fix(mgb/imperative): fix imperative code gen
GitOrigin-RevId: da9e8a280a
|
2 years ago |
Megvii Engine Team
|
c49d3070ba
|
refactor(imperative/ops): extends DnnOprCaller with template
GitOrigin-RevId: 402cba209a
|
2 years ago |
Megvii Engine Team
|
2d6476a416
|
feat(lite): add auto decide model inference format option
GitOrigin-RevId: fcbf945de5
|
3 years ago |
Megvii Engine Team
|
10a0349eca
|
feat(lite): add assert log for set_data_by_share
and set_data_by_copy. pylite network input is not
correct when input np is not continuous
GitOrigin-RevId: 1bdeae970a
|
2 years ago |
Megvii Engine Team
|
f5597d9a10
|
fix(mgb): make error infomation of input channel mismatch more readable
GitOrigin-RevId: 6f95260070
|
2 years ago |
Megvii Engine Team
|
38bd599911
|
fix(mgb): make error infomation of invalid MatMul more readable
GitOrigin-RevId: 96b922dd20
|
2 years ago |
Megvii Engine Team
|
e0d505e6bd
|
fix(mgb/dnn): fix bug that some cutlass file compile very slowly on SM86
GitOrigin-RevId: 91d7ac1927
|
2 years ago |
Megvii Engine Team
|
cc31b9db34
|
docs(mge/functional): fix vision docs
GitOrigin-RevId: f03567d8ae
|
2 years ago |
Megvii Engine Team
|
6f7649e935
|
docs(docstring): fix pad docstring
GitOrigin-RevId: 698d2f1c0d
|
2 years ago |
Megvii Engine Team
|
bf32d1e04f
|
docs(dataloader): update dataloader docstring
GitOrigin-RevId: 3e94a4bdf4
|
2 years ago |
Megvii Engine Team
|
4a32cc493a
|
docs(mge/data): update Dataset class docstring
GitOrigin-RevId: f08d818cf3
|
2 years ago |
Megvii Engine Team
|
70fc568224
|
docs(mge/data): update MNIST dataset docstring
GitOrigin-RevId: 536a46325f
|
2 years ago |
kagome1007
|
8fb062dfba
|
Merge pull request #468 from MegEngine/HuaHua404-patch-2
docs(Readme): add key features description
|
2 years ago |
kagome1007
|
e5ad3ea520
|
Merge pull request #469 from MegEngine/HuaHua404-patch-3
docs(readme): Add key features description
|
2 years ago |
Megvii Engine Team
|
58ba080d5f
|
feat(x86/rvv): make gi conv algo adapt to vv and vf model
GitOrigin-RevId: f29593be4d
|
2 years ago |
Megvii Engine Team
|
bd50e457ee
|
feat(x86/rvv): make MATRIX_MUL_GI_F32_4x12 and FP32_GEMV_MK4_GI
adapt to vv and vf model
GitOrigin-RevId: 691434c598
|
2 years ago |
Megvii Engine Team
|
5c3b4e9584
|
feat(x86/rvv): opt AlgoFP32WinogradF63_4x4_NCHW44
GitOrigin-RevId: 0cd0089982
|
2 years ago |
Megvii Engine Team
|
fa59a7b061
|
feat(x86/rvv): opt AlgoF32DirectNCHWNCHW44
and opt GiMaximumFloat32/GiMinimumFloat32 on x86
GitOrigin-RevId: 825021e867
|
2 years ago |
Megvii Engine Team
|
0d82e9b72b
|
feat(x86/rvv): opt FB_GI_F32_MK4_4x8
GitOrigin-RevId: 9e17de18b4
|
2 years ago |
Megvii Engine Team
|
3fbceb3a38
|
fix(mgb/version): fix nvinfer.h not found
GitOrigin-RevId: 981dd9a7d2
|
2 years ago |
Megvii Engine Team
|
a54d9cb9cd
|
feat(x86/rvv): opt FB_GI_F32_MK4_PACK_4x12 algo
GitOrigin-RevId: a80805c119
|
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
|
8fed114a51
|
fix(build): disable cupti temporarily
GitOrigin-RevId: d53cac6c1e
|
2 years ago |
Megvii Engine Team
|
19af26887e
|
fix(mge/tools): fix module_stats for duplicated module
GitOrigin-RevId: a15f17d616
|
2 years ago |
Megvii Engine Team
|
4cd4a38a7d
|
fix(mge/tools): fix network_visualize for op without out shapes
GitOrigin-RevId: fdde52c214
|
2 years ago |
Megvii Engine Team
|
7badcb72b6
|
build(wheel): copy libcupti.so
GitOrigin-RevId: bba5bf336f
|
2 years ago |
Megvii Engine Team
|
83fb1622a4
|
build(wheel): copy libcupti.so
GitOrigin-RevId: 8ff807012d
|
2 years ago |
Megvii Engine Team
|
6dfd5a4cd4
|
fix(win7): workaround for hang when progress exit on win7+32bit
GitOrigin-RevId: b49b0b230e
|
2 years ago |
Megvii Engine Team
|
ba1508e33a
|
fix(lite): fix exception bug for load and run
GitOrigin-RevId: 339a343a80
|
3 years ago |
Megvii Engine Team
|
603d09418e
|
fix(lite): fix layout transform bug in lar for testcase model
GitOrigin-RevId: c97b1c876b
|
2 years ago |
HuaHua404
|
8e48b17aa6
|
docs(README_CN): add key features description
|
2 years ago |
HuaHua404
|
0004291a28
|
docs(Readme): add key features description
|
2 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
|
247e2f59a4
|
feat(mgb/dnn): add modes that the output type is bool in elemwise
GitOrigin-RevId: fd0134fca2
|
3 years ago |
Megvii Engine Team
|
f386381080
|
fix(imperative): fix inplace operation of optim
GitOrigin-RevId: 2aaa71eb66
|
3 years ago |
Megvii Engine Team
|
9330929f26
|
feat(lite): add c_opr_init_interface for lar
GitOrigin-RevId: f7bcc27656
|
3 years ago |
Megvii Engine Team
|
16ba05a81b
|
fix(dnn): fix dnn nchwxx elemwise performance
GitOrigin-RevId: 5a715d7b2a
|
2 years ago |
huangxinda
|
9de8c122ad
|
feat(mge/third_party): add cudnn-frontend
|
2 years ago |
huangxinda
|
2b85948cd1
|
feat(mge/third_party): update flatbuffers version
|
2 years ago |
Megvii Engine Team
|
6c4c4ca6e6
|
Merge pull request #459 from Qsingle:fix_overflow_of_flops_calculate
GitOrigin-RevId: c1333e2089
|
2 years ago |
Megvii Engine Team
|
7b17c1180e
|
refactor(dnn): make cudnn_frontend work
GitOrigin-RevId: f089f93494
|
3 years ago |
Megvii Engine Team
|
35e9cc9845
|
feat(dnn/cuda): add cudnn frontend api
GitOrigin-RevId: 9b18a57893
|
3 years ago |
Megvii Engine Team
|
9be6d7b663
|
fix(lite): fix lite header
GitOrigin-RevId: e255597121
|
3 years ago |
Megvii Engine Team
|
ab8f6398d9
|
fix(test): make test install
GitOrigin-RevId: e38d6c5e9f
|
3 years ago |
Megvii Engine Team
|
99cfefbfe0
|
fix(test): fix test copybara
GitOrigin-RevId: 19b7bdf377
|
3 years ago |
Megvii Engine Team
|
0d7ace15c8
|
fix(mgb/dnn): suport fp16 for resize nhwc
GitOrigin-RevId: bb04d2a801
|
3 years ago |
Megvii Engine Team
|
cfed86f9af
|
feat(persistentcache): change file persistent cache with append model
GitOrigin-RevId: 7a427bdab4
|
3 years ago |