Megvii Engine Team
6bb6787d9a
feat(mge): add a tool which can analyze the file generated by compare_binary_iodump.py
GitOrigin-RevId: 9acab0a49f
4 years ago
Megvii Engine Team
43ef974842
feat(mgb): load_and_run can set both fastrun and reproducible
GitOrigin-RevId: 14e1de9952
4 years ago
Megvii Engine Team
c3f8cf04fa
feat(dnn): add conv_bwd_data and conv_bwd_filter accuracy shake check
GitOrigin-RevId: 4069e083d2
4 years ago
Megvii Engine Team
717b88e684
fix(mge/elemwise): fix problem that elemwise.mode is not comparable with string mode
GitOrigin-RevId: 82e39be0a9
4 years ago
Megvii Engine Team
9c17cfc45d
fix(imperative/ops): add check_dtype for Elemwise in infer_attrs
GitOrigin-RevId: c777855753
4 years ago
Megvii Engine Team
05550bc5b1
fix(mge/quantization): fix a compute problem
GitOrigin-RevId: 8aaedea959
4 years ago
Megvii Engine Team
813dd2f1c0
fix(mge/core): support non-Tensor args for `core.tensor.utils.get_device`
GitOrigin-RevId: 51dc03db9e
4 years ago
Megvii Engine Team
9f5bb42d78
perf(fastrun): cache persistent cache
GitOrigin-RevId: 21f7d4c19d
4 years ago
Megvii Engine Team
6ab1c55d2c
fix(mgb): fix fastrun workspace limit
GitOrigin-RevId: b9797eefcf
4 years ago
Megvii Engine Team
7af1ec66db
fix(tensor): del valid tensors when compnode finalizing
GitOrigin-RevId: bace1f2b51
4 years ago
Megvii Engine Team
696d2c2ef1
fix(mge/autodiff): check tensors to be attached
GitOrigin-RevId: c4f3f80876
4 years ago
Megvii Engine Team
984d85ca26
feat(mge/functional): argmin and argmax support negtive axis
GitOrigin-RevId: a1bd1102a6
4 years ago
Megvii Engine Team
87d6ff228d
fix(mge): fix dumping backward graph
GitOrigin-RevId: 430f110053
4 years ago
Megvii Engine Team
f31752d5cc
feat(mge/module): add __repr__ method for qat and quantized module
GitOrigin-RevId: 0d78cbab9a
4 years ago
Megvii Engine Team
9451a96147
test(mge/optimizer): update optimizer test to make sure grad not change
GitOrigin-RevId: e207672116
4 years ago
Megvii Engine Team
92e2ed6e2e
fix(mge/optimizer): fix optimizer update step inplace add grad
GitOrigin-RevId: d677d1ca6b
4 years ago
Megvii Engine Team
6972896933
fix(mge/utils): fix network multiple outputs issue
GitOrigin-RevId: d22e639cd3
4 years ago
Megvii Engine Team
f36e99d30b
fix(build): fix naive build
GitOrigin-RevId: 0050ff5d9c
4 years ago
Megvii Engine Team
516769391c
fix(mge/imperative): fix proxy graph comp_node finalize
GitOrigin-RevId: 22ad0d0fc6
4 years ago
Megvii Engine Team
13e6ea349d
feat(imperative/opr): rebase rng refactoring to dev & add python module
GitOrigin-RevId: ee5984c52d
4 years ago
Megvii Engine Team
cded8ef1a6
feat(imperative): add rng opdef
GitOrigin-RevId: b62dcea7f5
4 years ago
Megvii Engine Team
40bab1ed66
feat(log): opt log, enable mgb sdk log at opt build
more info: 16cd674c56
* change MGE_OVERRIDE_LOG_LEVEL to RUNTIME_OVERRIDE_LOG_LEVEL
* use ::std::getenv not MGB_GETENV for special ENV
GitOrigin-RevId: ee0f9c0f72e627c331c00100f6a21adc927081df
4 years ago
Megvii Engine Team
c3a1ac3dcf
feat(mge/tools): module_status-add-functions
BREAKING CHANGE:
GitOrigin-RevId: ced3da3a12
4 years ago
Megvii Engine Team
05e4c8266d
fix(mge/module): fix a mistaken in BN docs
GitOrigin-RevId: 086a70649e
4 years ago
Megvii Engine Team
f6e66ec69c
fix(data): fix the sampling scale in Lighting
GitOrigin-RevId: d0a1ca50e2
4 years ago
Megvii Engine Team
00b48dfe7f
Revert "perf(opr): use pin mem for param_pack_concat"
This reverts commit 95a0997372
GitOrigin-RevId: 12ab59b8aa
4 years ago
Megvii Engine Team
e18afa0b05
feat(mge/module): python wrapper for conv_transpose3d
GitOrigin-RevId: 61097b8713
4 years ago
Megvii Engine Team
fdf7006ba7
fix(imperative): add __init__.py for tools
GitOrigin-RevId: 11ec38a4f1
4 years ago
Megvii Engine Team
f16c9eb9d7
refactor(mgb): simplify fast run proccess with storing algo desc instead of algo name
GitOrigin-RevId: 04b86fde50
4 years ago
Megvii Engine Team
3e172f3b73
feat(lite): add cmake build for rknn and mge in lite
GitOrigin-RevId: c6c77a4b61
4 years ago
Megvii Engine Team
5d637d0723
refactor(mgb): code refactor of fast run
GitOrigin-RevId: 2c4b8e06bb
4 years ago
Megvii Engine Team
f6bd4f59f7
fix(mgb): fix attribute check compatibility when profiling and read_from_cache in fast run
GitOrigin-RevId: b865d467b0
4 years ago
Megvii Engine Team
0a86a07096
fix(mgb/dnn): fix cub potential issues
Wrap cub with CUB_NS_PREFIX and remove dependency on Thrust to avoid potential linking issues
GitOrigin-RevId: 53893b0a39
4 years ago
Megvii Engine Team
282dfc6232
refactor(imperative): alloc enum type class on heap
GitOrigin-RevId: d2b2acea22
4 years ago
Megvii Engine Team
1e6ef3771f
feat(mgb/dnn): add accuracy shake checker
GitOrigin-RevId: 0bb52078a1
4 years ago
Megvii Engine Team
a5a29826dc
feat(mgb/dnn): add accuracy_depend_on_batch attribute
GitOrigin-RevId: 2be1db43d0
4 years ago
Megvii Engine Team
4b141f8de4
fix(mgb): add usable-depend-on-shape attr
GitOrigin-RevId: 3a14fa6b6f
4 years ago
Megvii Engine Team
15b647aee2
fix(externcopr): check loader imp dynmaic param
GitOrigin-RevId: 5f38c120e3
4 years ago
Megvii Engine Team
5c131d2a74
fix(device): fix get_device_count for xpu
GitOrigin-RevId: 6c77f507f3
4 years ago
Megvii Engine Team
9eb4ba8836
perf(tensor): always copy to pinned memory before h2d
GitOrigin-RevId: f945f2435f
4 years ago
Megvii Engine Team
4d9073fcba
test(dist): remove redundant assert for reduce_sum and gather
GitOrigin-RevId: bc7393bb63
4 years ago
Megvii Engine Team
3bd0df8e12
fix(rocm): enable var_releaser for rocm
GitOrigin-RevId: a42185aa6f
4 years ago
Megvii Engine Team
21c6c4371d
perf(opr): use pin mem for param_pack_concat
GitOrigin-RevId: 95a0997372
4 years ago
Megvii Engine Team
1a7112997c
feat(opr-mm): add backend argument for remote send/recv
GitOrigin-RevId: 841a0e45ab
4 years ago
Megvii Engine Team
69a146c8c3
build(rocm): support rocm-3.9
GitOrigin-RevId: 85f8911736
4 years ago
Megvii Engine Team
dee5a10af9
feat(distributed): auto detect device and backend when init group
GitOrigin-RevId: 90be2d5b4d
4 years ago
Megvii Engine Team
1bec737d1a
feat(distributed): support distributed opr for rocm
GitOrigin-RevId: 4840100d07
4 years ago
Megvii Engine Team
a31b7c6ee3
build(rocm): partially support hcc compilation
GitOrigin-RevId: ca9f1f8e8e
4 years ago
Megvii Engine Team
621ae0a1e8
fix(dnn): replace kernel launch syntax with macro for hcc
GitOrigin-RevId: f9e69d4825
4 years ago
Megvii Engine Team
78fff72a95
feat(dnn): add param_pack for rocm
GitOrigin-RevId: 2180504c71
4 years ago