Megvii Engine Team
|
227ec04bd3
|
fix(mge/test): fix pytest failed issue
GitOrigin-RevId: 96df470406
|
4 years ago |
Megvii Engine Team
|
b9a774dd19
|
chore(deps): add pybind11
|
4 years ago |
Megvii Engine Team
|
6b380e8965
|
feat(mge/imperative): run oss test and restore cmake list build items
GitOrigin-RevId: 11411b6964
|
4 years ago |
Megvii Engine Team
|
0380811218
|
feat(dnn/arm_common): add nchw44 8x8x16 stride1 stride2
2x2 3x3 5x5 7x7 directconv
GitOrigin-RevId: 3710182af1
|
4 years ago |
Megvii Engine Team
|
2dbe8194ad
|
fix(mge/opr): fix reduction static infer value
GitOrigin-RevId: 6623d4719a
|
4 years ago |
Megvii Engine Team
|
c20d4cc6dc
|
feat(dnn): fix opt pass nchw44 can not dump resnet
GitOrigin-RevId: 28e5c37f53
|
4 years ago |
Megvii Engine Team
|
3dbac4f47f
|
feat(mge): add atlas_subgraph module
GitOrigin-RevId: 11530383c0
|
4 years ago |
Megvii Engine Team
|
00ef677249
|
fix(mgb): remove internal for cambricon and atlas
GitOrigin-RevId: 861e349eb4
|
4 years ago |
Megvii Engine Team
|
aeffcd5897
|
feat(dnn/cuda): integrate cutlass nchw32 tensorcore convolution
GitOrigin-RevId: 9d6c48ed99
|
4 years ago |
wkcn
|
19568f67e9
|
chore(gitmodule/acknowledgments): update mkldnn link and acknowledgments
|
4 years ago |
Megvii Engine Team
|
9e5e32dee2
|
fix(dnn): restore opr_param_defs.py
GitOrigin-RevId: b92747cad3
|
4 years ago |
Megvii Engine Team
|
e13fb7b99b
|
fix(mge/quantization): fix apply observer bug which get qparam before call observer
GitOrigin-RevId: 7fd6333683
|
4 years ago |
Megvii Engine Team
|
d334b229b0
|
feat(imperative): add nms opr wrapper
GitOrigin-RevId: d92241a234
|
4 years ago |
Megvii Engine Team
|
bca00f2e22
|
fix(dnn): midout at where neccessary in megdnn
GitOrigin-RevId: 191334bd96
|
4 years ago |
ChaiMind
|
105e44504a
|
chroe(.gitignore): add more ignore rules for git
|
4 years ago |
ChaiMind
|
2d81e03847
|
fix(git): make git version check work
|
4 years ago |
Megvii Engine Team
|
b91bfd59cf
|
fix(sdk/load-and-run): fix load-and-run error after adding testcases on gpu
GitOrigin-RevId: 32429d8543
|
4 years ago |
Megvii Engine Team
|
bc2dfcbb53
|
refactor(cmake): clear cmakelist file may caused by some commit conflict fix error
GitOrigin-RevId: ac5acfdf4c
|
4 years ago |
Megvii Engine Team
|
a1e6720756
|
feat(dnn): enable bool comparison
GitOrigin-RevId: 735693b81e
|
4 years ago |
Megvii Engine Team
|
8aa34e4a5d
|
feat(imperative): add advance indexing with bool
also fix bug to deal with empty tensor in np2tensor
GitOrigin-RevId: 2bd8152c90
|
4 years ago |
Megvii Engine Team
|
101b58d1ca
|
fix(dnn): enable bool input to cond_take
GitOrigin-RevId: 532fa7c073
|
4 years ago |
Megvii Engine Team
|
eb0cc59dcf
|
feat(windows/python_whl/cuda): make windows MORE HAPPY for build whl with cuda
GitOrigin-RevId: 9e7d7d4167
|
4 years ago |
Megvii Engine Team
|
4a178a8dba
|
feat(windows/cuda/cmake): support cmake cuda build on windows
GitOrigin-RevId: 4d9832e559
|
4 years ago |
Megvii Engine Team
|
1915593b6c
|
fix(version_depend): add a fake version on dev
make dev useable on all megskull version
GitOrigin-RevId: 8c4739d470
|
4 years ago |
Megvii Engine Team
|
6aade1336d
|
fix(dnn/fallback): disable im2col/conv1x1/conv1x1_gemv Quantized8Asymm in x86
GitOrigin-RevId: b094634254
|
4 years ago |
Megvii Engine Team
|
42124ab7cf
|
fix(mgb/cmake): fix cmake linker flags issue
GitOrigin-RevId: ed41479402
|
4 years ago |
Megvii Engine Team
|
991286e81d
|
feat(macos/ci): add macos ci
GitOrigin-RevId: b20a514f8c
|
4 years ago |
Megvii Engine Team
|
56381f808b
|
fix(dnn/arm): use vcvtq_f32_s32 for all arm code
GitOrigin-RevId: 27effe7d24
|
4 years ago |
Megvii Engine Team
|
1173205726
|
fix(gopt): nchw_nchwxx useable and opt pass use nchw_nchwxx_valid
GitOrigin-RevId: 60942aca5b
|
4 years ago |
Megvii Engine Team
|
eb18eba87d
|
fix(gopt): fix nchw44 nchw44_dot gopt test
GitOrigin-RevId: 06b38dcd30
|
4 years ago |
Megvii Engine Team
|
40e79e9dab
|
fix(dnn/x86): fix x86 matrix usable ignore format
GitOrigin-RevId: 40fe508aca
|
4 years ago |
Megvii Engine Team
|
eab7ab0530
|
fix(gopt): gen nchw_nchw44 when kernel is optimized
GitOrigin-RevId: 89083be200
|
4 years ago |
Megvii Engine Team
|
777f3ea970
|
refactor(gopt): format code
GitOrigin-RevId: 9d5c87000f
|
4 years ago |
Megvii Engine Team
|
b44e0549b1
|
fix(mgb): add method to check local cuda env and refactor the way to get cuda include and lib dir
GitOrigin-RevId: 63f7302ce1
|
4 years ago |
Megvii Engine Team
|
30ce3c60bd
|
Revert "fix(mgb/opr): change EQ opr's backward_graph to nullptr instead of InvalidGrad"
This reverts commit 6355337641 .
GitOrigin-RevId: 6897f6d45d
|
4 years ago |
Megvii Engine Team
|
14e71b551b
|
feat(imperative): add helper for dnn opr caller
GitOrigin-RevId: 828cd69803
|
4 years ago |
Megvii Engine Team
|
2272abe18d
|
fix(mgb/fallback): disable nchw44 in conv1x1 and im2col in x86
GitOrigin-RevId: 603d2eb94a
|
4 years ago |
Megvii Engine Team
|
230ab45a1e
|
fix(mgb/naive): fix naive convolution no dispatch kernel in handle
GitOrigin-RevId: 4038fe23a4
|
4 years ago |
Megvii Engine Team
|
22853fa20c
|
feat(mge/quantization): add `mapping` parameter for custom modules
GitOrigin-RevId: a4de4261d0
|
4 years ago |
Megvii Engine Team
|
6e70fa7a11
|
feat(dnn/arm): add fp32 asm gemm for a53 a55 and i8i8i16 gemm for a72 a53
GitOrigin-RevId: a049c33f2b
|
4 years ago |
Megvii Engine Team
|
dbaf84b0ef
|
feat(imperative): add cond_take opr
GitOrigin-RevId: 5272e6fa71
|
4 years ago |
Megvii Engine Team
|
df356635b7
|
fix(mgb/fallback): delete im2col duplicate code and fix nchw44 usable
GitOrigin-RevId: 1aa250e9e7
|
4 years ago |
Megvii Engine Team
|
4a2270834f
|
fix(mgb/fallback): fix conv1x1 and conv1x1_gemv nchw44 usable
GitOrigin-RevId: 90aa75d51e
|
4 years ago |
Megvii Engine Team
|
b778d22523
|
feat(mgb/fallback): add conv1x1_gemv, conv1x1 and im2col 8x8x16/8x8x32 support bias
GitOrigin-RevId: 3d97fedc8f
|
4 years ago |
Megvii Engine Team
|
c357db0134
|
feat(mgb/arm_common): add 8x8x16 nchw44 max pooling
GitOrigin-RevId: ed460adb7a
|
4 years ago |
Megvii Engine Team
|
7f5f375fda
|
feat(dnn/arm): add armv7 nchw_nchw44 3x3s2 asm kernel
GitOrigin-RevId: 50ce91e41d
|
4 years ago |
Megvii Engine Team
|
b7d5fa7e64
|
fix(sdk/load_and_run): fix misuse std::string::substr
GitOrigin-RevId: f45a84696a
|
4 years ago |
Megvii Engine Team
|
1bce857cb8
|
fix(mgb/opr-mm): use comp_node of config as default in CollectiveComm
GitOrigin-RevId: 6b43c9fc93
|
4 years ago |
Megvii Engine Team
|
27205461ae
|
feat(mgb/opr-mm): add register info cache for multi-machine oprs
GitOrigin-RevId: d5ae3c5a7c
|
4 years ago |
Megvii Engine Team
|
a7ff580e54
|
feat(mge/utils): add net stats to calculate parameters and flops
GitOrigin-RevId: a77f89e24b
|
4 years ago |