Megvii Engine Team
f354724220
fix(ci/megdnn_test/megbrain_test): split some
testcase, which caused by timeout
GitOrigin-RevId: ffed9d7820
5 years ago
Megvii Engine Team
ff44f14f80
Merge branch 'master' into dev
GitOrigin-RevId: 4e661180b4
5 years ago
Megvii Engine Team
1af4863179
fix(dnn/fallback): fix conv1x1's is_preferred method
GitOrigin-RevId: 6d34080d25
5 years ago
Megvii Engine Team
206521cd33
fix(mge/quant): avoid updating the scale and zero point in eval mode
GitOrigin-RevId: dfb08cf701
5 years ago
Megvii Engine Team
c9986df596
feat(dnn/arm): add fp32 nchw_nchw44 conv
GitOrigin-RevId: f19fe892d9
5 years ago
Megvii Engine Team
ca855d8d21
feat(dnn/fallback): fuse im2col and packb 4x4x16
GitOrigin-RevId: 123920899d
5 years ago
Megvii Engine Team
4f77509ea6
feat(mgb/opr): allow empty ImmutableTensor
Fixes MGE-675.
GitOrigin-RevId: c6771740fc
5 years ago
Megvii Engine Team
6a7e7ce1b1
fix(mge/cgtools): make get_dep_vars accept symbolvar input correctly
GitOrigin-RevId: 8bf00a3006
5 years ago
Megvii Engine Team
f19646b51f
feat(dnn/arm_common/elemwise): elemwise ternary support chw44
GitOrigin-RevId: ef19a636ba
5 years ago
Megvii Engine Team
3d9d4b9beb
fix(sdk/load_and_run): fix ciruclar dependency in DataPaser::Brace
GitOrigin-RevId: d07f8c19b2
5 years ago
Megvii Engine Team
3bd699fdfd
fix(opr): fix int8 winograd preprocess output dtype mismatch
GitOrigin-RevId: ede80d5a45
5 years ago
Megvii Engine Team
2ae9fdef40
feat(dnn/arm): add arm common nchw44 avg pooling
GitOrigin-RevId: 25eab33e14
5 years ago
Megvii Engine Team
0293d58ade
feat(mge): add bfloat16 support
GitOrigin-RevId: a942ce6791
5 years ago
Megvii Engine Team
c3d5b61f1c
feat(mgb/opr): add property params for python operator class
GitOrigin-RevId: af6da0e0ac
5 years ago
Megvii Engine Team
81915f6bbf
feat(sdk/load_and_run): support npy/json/npy/ppm/pgm as input
GitOrigin-RevId: f0aa7abe85
5 years ago
Megvii Engine Team
df8931b6cd
feat(dnn/arm): add padding support for nchw44 arm pooling and opt code
GitOrigin-RevId: f125004e1f
5 years ago
Megvii Engine Team
07dd6b6c4e
feat(dnn/arm): add arm nchw44 filter5x5 strdie1 and stride2 max pooling
GitOrigin-RevId: f84c32f922
5 years ago
Megvii Engine Team
15cca8f9d6
feat(dnn/arm): add arm nchw44 filter4x4 strdie1 and stride2 max pooling
GitOrigin-RevId: 3a4da20c59
5 years ago
Megvii Engine Team
ab401aba14
feat(dnn/arm): add arm nchw44 filter2x2 strdie1 and stride2 max pooling
GitOrigin-RevId: 42d144a813
5 years ago
Megvii Engine Team
b336db65b5
feat(dnn/arm): add arm nchw44 filter3x3 strdie1x1 max pooling
GitOrigin-RevId: b0d54d38ad
5 years ago
Megvii Engine Team
9837bc00a4
feat(dnn/arm): add arm nchw44 filter3x3 strdie2x2 max pooling
GitOrigin-RevId: 02c7a8405c
5 years ago
Megvii Engine Team
e109ae919c
fix(mgb/gopt): fix float32 to float16 opt pass
GitOrigin-RevId: d828512e44
5 years ago
Megvii Engine Team
1255c9f13d
feat(mge/opr): add opr remap in opencl and naive
GitOrigin-RevId: 4540788660
5 years ago
Megvii Engine Team
03d6805870
fix(windows/dnn/ci): fix cpu x86_windows_xp flags build issue
*as xp 32 mkl crash issue after "11.1.4 (build date 20140806)"
we can not upgrade intel_mkl_windows_xp_archive to
MKL 2019.0.1 (build date 20180928), so we make a difference between
xp with linux/macos/windows_no_xp
GitOrigin-RevId: 97b049d7893fec217aaa837f0a6909991b788cf0
5 years ago
zjd1988
d34e385dd7
assert locator.device < sd.MAX_NR_DEVICE when bool(avaiable_node) = true
Signed-off-by: zjd1988 <942002795@qq.com>
5 years ago
Megvii Engine Team
502d6b893d
build(manylinux2010): add missing libatomic depedency
GitOrigin-RevId: 8f85ace96c
5 years ago
Megvii Engine Team
a795aa700a
fix(mge/data): fix wrong class name in voc
GitOrigin-RevId: 229038bcfa
5 years ago
Megvii Engine Team
ae506ecfce
docs(mge/module): fix momentum equation in batch norm doc
GitOrigin-RevId: 747ae1646a
5 years ago
Megvii Engine Team
1c4a64b2af
test(megdnn): skip fp16 test if compute capability less than 60
GitOrigin-RevId: 3377007b87
fix(dnn): fix include file redefined error
GitOrigin-RevId: 9d52f7ba50
5 years ago
luzzyzhang
fd113acb8b
ci(github): change docker `shm-size` to fix `test_distributed` error
5 years ago
Xinran Xu
3f4499d157
chore(release): bump version
5 years ago
Xinran Xu
f69a4371f9
build(megray): update megray commit
5 years ago
Megvii Engine Team
429cca8e2b
docs(mge/hub): refine docstring in hub module
GitOrigin-RevId: 2f73c1242b
5 years ago
Megvii Engine Team
955095c1a9
docs(mge/utils): refine docstring in utils module
GitOrigin-RevId: b2bc163e15
5 years ago
Megvii Engine Team
b2ed2630f7
Revert "fix(mge/functional): rm useless arguments"
This reverts commit 8e4f25bfd8
.
GitOrigin-RevId: df696ab8a2
5 years ago
Megvii Engine Team
819a4f1c9b
feat(cross_build): add cmake cross build ci
GitOrigin-RevId: ada9554bc2
5 years ago
Megvii Engine Team
d23fec06cb
fix(mge/functional): rm useless arguments
group and compute_mode are not used by local_conv2d
GitOrigin-RevId: 8e4f25bfd8
5 years ago
Megvii Engine Team
3b08bd9eb9
fix(dnn/fallback): fix im2col thread safe problem
GitOrigin-RevId: f9f82d8c88
5 years ago
Megvii Engine Team
3ef308e71c
ci(copybara): fix copybara of arm
GitOrigin-RevId: 2aa113ef47
5 years ago
Megvii Engine Team
8659876794
docs(mge/functional): fix doctest
GitOrigin-RevId: cd23860ab8
5 years ago
Megvii Engine Team
4500faf1ea
fix(python_module/mge): one_hot: no default value for num_classes
GitOrigin-RevId: c4a5310880
5 years ago
Megvii Engine Team
b9cf01718a
fix(dnn/x86): replace suonds bad name
GitOrigin-RevId: b8f0f03bb0
5 years ago
Megvii Engine Team
8c110c3942
feat(mge/quantization): add quantization interface
GitOrigin-RevId: 4fe1233ec3
5 years ago
Megvii Engine Team
384e75781d
build(mge): put LICENSE file into proper directory in wheel
GitOrigin-RevId: 77f3a6968b
5 years ago
Megvii Engine Team
5697fa169a
feat(dnn): conv1x1 nchw44 support
GitOrigin-RevId: 29b41ff460
5 years ago
Megvii Engine Team
a323f1a449
feat(dnn/arm): add nchw nchw44 first conv
GitOrigin-RevId: 563f57844f
5 years ago
Megvii Engine Team
814a427810
feat(mge/module): add `with_parent` argument in `_flatten`
GitOrigin-RevId: 1c88559ece
5 years ago
Megvii Engine Team
2a1e0624ab
docs(mge/functional): fix doctest
GitOrigin-RevId: 45b6e1147e
5 years ago
Megvii Engine Team
5d32db8a10
docs(mge/functional): fix docs error and refine description
GitOrigin-RevId: 62da6f0d60
5 years ago
Megvii Engine Team
f71251fd0c
docs(mge/functional): refine the docstring of several apis
GitOrigin-RevId: 55f61be2a8
5 years ago