Megvii Engine Team
ae6ff2c5a6
feat(mgb/gopt): add opt pass for nchw64 layout transform
GitOrigin-RevId: adede7cef6
4 years ago
Megvii Engine Team
63a9bd30a8
feat(mgb/gopt): add an opt pass for padding channels to enable fast int8/int4 support on GPU
GitOrigin-RevId: 94c719bb5c
4 years ago
Megvii Engine Team
858261af1f
fix(python_module): fix conversion between numpy-ndarray and mgb tensor for qint4 and quint4
GitOrigin-RevId: 7450c4f25e
4 years ago
Megvii Engine Team
3b9b87809d
refactor(dnn): refactor lowbit tensor format
GitOrigin-RevId: b646dc085b
4 years ago
Megvii Engine Team
2d6827c168
fix(mgb/windows): temporary workround on cuda-windows python exit
code(127), as windows cuda driver unloading before atexit function
may remove this after upgrade cuda runtime
GitOrigin-RevId: cac37ca3dd
4 years ago
Megvii Engine Team
d2e33af52f
fix(mgb): fix wrong set of strategy in lar
GitOrigin-RevId: 5c1f7c669f
4 years ago
Megvii Engine Team
8b7d8d290b
fix(core): fix json dump when weight preprocess
GitOrigin-RevId: 6cd882b10d
4 years ago
Megvii Engine Team
ec65e1f9ba
fix(build/windows): fix windows build:
* compat clang-cl 11 build at windows env
* fix cuda/cudnn/trt copy env build failed on windows
GitOrigin-RevId: 7fe2d2c0dc
4 years ago
Megvii Engine Team
b06b589960
feat(mgb): get static graph memory info
GitOrigin-RevId: f31745f8df
4 years ago
Megvii Engine Team
3591ef1f6a
fix(mgb): fix conv cudnnconvbackwarddata algo witch is not shake
GitOrigin-RevId: 379bfbe376
4 years ago
Megvii Engine Team
1525a02530
feat(mge/module): add python wrapper for unfold
GitOrigin-RevId: 562103186f
4 years ago
Megvii Engine Team
13b15fb08c
feat(megbrain): add correlation opr
GitOrigin-RevId: 6d44598891
4 years ago
Megvii Engine Team
9b69a02fb2
fix(imperative/tensor): init m_offset when constructing a Tensor with DeviceTensorND
GitOrigin-RevId: b340e27c47
4 years ago
Megvii Engine Team
36b1ba052f
fix(mgb/dnn): fix cudnn8.0.4 convbias with z
GitOrigin-RevId: 09453d8a12
4 years ago
Megvii Engine Team
6ab1c55d2c
fix(mgb): fix fastrun workspace limit
GitOrigin-RevId: b9797eefcf
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
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
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
f16c9eb9d7
refactor(mgb): simplify fast run proccess with storing algo desc instead of algo name
GitOrigin-RevId: 04b86fde50
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
1e6ef3771f
feat(mgb/dnn): add accuracy shake checker
GitOrigin-RevId: 0bb52078a1
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
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
1bec737d1a
feat(distributed): support distributed opr for rocm
GitOrigin-RevId: 4840100d07
5 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
c9348b162f
build(cuda): link to cuda_stub
GitOrigin-RevId: 073d3313cf
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
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
2d18074a70
fix(mgb): fix spell error
GitOrigin-RevId: acae00e0a5
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
8494a1529e
chore(scripts): clarify and fix default value of bit combined enum
GitOrigin-RevId: 3716bf9bb5
4 years ago
Megvii Engine Team
acf2860341
fix(cmake): export more sym caused by we change
exception namespace when disable rtti
GitOrigin-RevId: c42f5adffd
4 years ago
Megvii Engine Team
aed681d325
feat(imperative/utils): optimize the naming rules
GitOrigin-RevId: 329bac640a
4 years ago
Megvii Engine Team
04b1a45af4
fix(dnn): fix cudnn crash when finalize called after cudnn dtor
GitOrigin-RevId: b0ad639921
4 years ago
Megvii Engine Team
14a089c49d
fix(dnn): change ci to cudnn804, reopen testcase
GitOrigin-RevId: 90713a801b
4 years ago
Megvii Engine Team
c338e876ec
refactor(mgb/dnn): add negative attribute for algo
GitOrigin-RevId: 88b1ce94a5
4 years ago
Megvii Engine Team
ec1a99acc2
refactor(mgb/dnn): replace reproducible with attribute
GitOrigin-RevId: d49015714c
4 years ago
Megvii Engine Team
6af0299c3f
test(mge/ioremote): add REQUIRE_GPU for ioremote opr test
GitOrigin-RevId: 31ec8982c4
4 years ago
Megvii Engine Team
bab0caab72
fix(mgb/opr): add layout constraint on the input of Cumsum
GitOrigin-RevId: d2d108e278
4 years ago
Megvii Engine Team
94401ce44a
chore(dotprod): dotprod is enabled by default on the android platform
GitOrigin-RevId: d412108732
4 years ago
Megvii Engine Team
1310ad4926
feat(dnn): add memory manager for enflame
GitOrigin-RevId: 72a810d959
5 years ago
Megvii Engine Team
2fd3fa8834
feat(cmake): update for enflame cmake compile
GitOrigin-RevId: 3c3c6b3462
5 years ago