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 |
Megvii Engine Team
|
6de3e4baa3
|
refactor(mgb/opr): make trt batch flag only depend on inputs dimension
GitOrigin-RevId: f2f1a10762
|
4 years ago |
Megvii Engine Team
|
ce610ca31a
|
fix(mgb): fix attribute uncomplete filter when get_profile_result_from_cache in fast run
GitOrigin-RevId: 599437ad72
|
4 years ago |
Megvii Engine Team
|
c992afa960
|
test(mge/module): add module reset attribute test
GitOrigin-RevId: 6c9adc4a70
|
4 years ago |
Megvii Engine Team
|
93bfda5191
|
fix(mge/module): fix module attribute update mistake
GitOrigin-RevId: 842a8fcb6e
|
4 years ago |
Megvii Engine Team
|
c9348b162f
|
build(cuda): link to cuda_stub
GitOrigin-RevId: 073d3313cf
|
4 years ago |
Megvii Engine Team
|
33ab2454b9
|
fix(whl): support wheel version without sdk info,
which caused by python official PYPI do not support
LOCAL_VERSION; sync lite setup.py
GitOrigin-RevId: 7e35ceb5f1
|
4 years ago |
Megvii Engine Team
|
8163ed157d
|
fix(dnn/cuda): fix cutlass matmul splitk limit
GitOrigin-RevId: fc9a7c638c
|
4 years ago |
Megvii Engine Team
|
933188aa93
|
feat(functional/nn): support F.warp_perspective with `mat_idx`
GitOrigin-RevId: 66910c8bd8
|
4 years ago |
Megvii Engine Team
|
8585aa61ad
|
fix(mgb): fix fast run crash when profile heuristic strategy
GitOrigin-RevId: 6046a2db0c
|
4 years ago |
Megvii Engine Team
|
ef9aa80074
|
fix(mgb/dnn): fix cuda naive matmul algo
GitOrigin-RevId: 79c9bba73b
|
4 years ago |
Megvii Engine Team
|
fe99cdc794
|
feat(interpreter): add dynamic sublinear
GitOrigin-RevId: 7de54fe743
|
4 years ago |
Megvii Engine Team
|
62c394ca91
|
feat(cuda/comp_node): enable to query adjacent free blocks size
GitOrigin-RevId: 0cd1d202dd
|
4 years ago |
Megvii Engine Team
|
fd61f09540
|
feat(cuda/comp_node): enable to directly query memory status
GitOrigin-RevId: 23c1af9a55
|
4 years ago |
Megvii Engine Team
|
a3ab9f1d41
|
feat(blob_manager): make all memory allocation go through blob manager
GitOrigin-RevId: f79155e5c3
|
4 years ago |
Megvii Engine Team
|
3ea00ebab1
|
fix(cmake): fix cmake build failed at soft link workspace
GitOrigin-RevId: 78b50a4f3a
|
4 years ago |
Megvii Engine Team
|
a81085221a
|
refactor(imperative): refactor enum param type caster
GitOrigin-RevId: 1aae07f143
|
4 years ago |
Megvii Engine Team
|
dcff115ef8
|
fix(distributed/launcher): fetch data early to avoid dead lock in launcher
GitOrigin-RevId: 9abcc956ef
|
4 years ago |
Megvii Engine Team
|
7e22e9f06e
|
feat(optimzer): add AdamW
GitOrigin-RevId: e608b5d5b9
|
4 years ago |
Megvii Engine Team
|
2d18074a70
|
fix(mgb): fix spell error
GitOrigin-RevId: acae00e0a5
|
4 years ago |
Megvii Engine Team
|
65af9cfcdc
|
refactor(mge): use lower case for default string parameters in functional and module
GitOrigin-RevId: dbc1f27ff7
|
4 years ago |
Megvii Engine Team
|
d0aa9b41ee
|
refactor(mge/functional): move nvof to vision, compatible with old usage
GitOrigin-RevId: cd9d9b4f5a
|
4 years ago |
Megvii Engine Team
|
ff755451d2
|
refactor(mgb): move algo's name from info to desc and delete some algo's unnecessary param() method
GitOrigin-RevId: 144ff547d1
|
4 years ago |
Megvii Engine Team
|
a437ec8e88
|
fix(src/gopt): add replace func of typecvt opr for nhwcd4 pass
GitOrigin-RevId: 801eb1dab3
|
4 years ago |
Megvii Engine Team
|
b1baee60f5
|
feat(imperative/utils): add optimize-for-inference interface for opgraph
GitOrigin-RevId: 9f93f82190
|
4 years ago |
Megvii Engine Team
|
86598c8227
|
fix(mge/utils): filter out parameter "arg_names" and "output_name" in network.dump
GitOrigin-RevId: 408f52ad2d
|
4 years ago |