Megvii Engine Team
|
71c2f61254
|
feat(dnn/cuda): add relayout format to support layout transform between NCHW and NCHW64
GitOrigin-RevId: 1445ecfabe
|
4 years ago |
Megvii Engine Team
|
df009e89e1
|
feat(dnn/cuda): add cuda conv bias impls for NCHW format tensors with qint4 data type
GitOrigin-RevId: a0a08cf42c
|
4 years ago |
Megvii Engine Team
|
ed92207585
|
feat(dnn/cuda): add conv bias impl for int4 data type using sass language
GitOrigin-RevId: ae3d3e1c98
|
4 years ago |
Megvii Engine Team
|
52b55564d7
|
refactor(dnn/cuda): refactor reorder filter and bias kernel to support conv imma with data type s4
GitOrigin-RevId: 6827b73770
|
4 years ago |
Megvii Engine Team
|
d2673c5abf
|
fix(ci/windows): add windows cuda test
GitOrigin-RevId: 706be83032
|
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
|
517cc6846a
|
ci(gitlab-ci): add inline lineno checking in copybara linter
GitOrigin-RevId: 56c5068009
|
4 years ago |
Megvii Engine Team
|
23032f50f2
|
feat(dnn/cuda): support float16 for index_incr_multi_axis_vec
GitOrigin-RevId: c2ae93d568
|
4 years ago |
Megvii Engine Team
|
938944027d
|
fix(mgb/dnn): fix cudnn8 convbias
GitOrigin-RevId: 0fdbfd258c
|
4 years ago |
Megvii Engine Team
|
5427a67c96
|
fix(cmake/subdirectory): fix project import by other sdk by add-subdirectory
notice: if you are cross build, please follow step
----
define SRC_DIR you sdk root dir
SRC_DIR=$($READLINK -f "`dirname $0`")
source $SRC_DIR/megbrain/scripts/cmake-build/utils/utils.sh
build_flatc $SRC_DIR/megbrain false
your build code
----
GitOrigin-RevId: ad97860f5f
|
4 years ago |
Megvii Engine Team
|
241b35a697
|
refactor(ops): remove BackwardGraph op
GitOrigin-RevId: eda20e5760
|
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
|
8a918717c0
|
feat(mgb): add megengine inference
GitOrigin-RevId: 6ffec6b418
|
4 years ago |
Megvii Engine Team
|
379a28f9b0
|
feat(mgb): add load_and_run option
GitOrigin-RevId: b383beb886
|
4 years ago |
Megvii Engine Team
|
b06b589960
|
feat(mgb): get static graph memory info
GitOrigin-RevId: f31745f8df
|
4 years ago |
Megvii Engine Team
|
0cf4ff702e
|
perf(dist): add fastpath for bcast params
GitOrigin-RevId: aa40b3cd72
|
4 years ago |
Megvii Engine Team
|
5e54821845
|
docs(syncbn): complete syncbn document
GitOrigin-RevId: 9b8d546f0b
|
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
|
e417798f9e
|
fix(mge): correct pytype when calling apply from python
GitOrigin-RevId: 6abfa06ada
|
4 years ago |
Megvii Engine Team
|
c4048519c6
|
chore(mge/version): change dev version to 1.5
GitOrigin-RevId: 4058da1f98
|
4 years ago |
Megvii Engine Team
|
3352fe7003
|
refactor(dtr): import dtr as submodule
GitOrigin-RevId: abecd0f176
|
4 years ago |
Megvii Engine Team
|
a7581986bc
|
perf(dist): speed up bcast_val
GitOrigin-RevId: 21c4123b09
|
4 years ago |
Megvii Engine Team
|
1525a02530
|
feat(mge/module): add python wrapper for unfold
GitOrigin-RevId: 562103186f
|
4 years ago |
Megvii Engine Team
|
d07ef83aa8
|
feat(mge/tools): add bash_completion for user to hint script
BREAKING CHANGE:
fix(mge/tools): change help script for other shell
BREAKING CHANGE:
fix(mge/tools): fix double -h error
BREAKING CHANGE:
fix(mge/tools): add support for _ parameter
BREAKING CHANGE:
GitOrigin-RevId: 537c3a2aae
|
4 years ago |
Megvii Engine Team
|
efc7f04fd3
|
refactor(mge/jit): using static global_enable for apply ctx insted of global variable
GitOrigin-RevId: dd82b53faf
|
4 years ago |
Megvii Engine Team
|
750bde9513
|
refactor(mge/jit): remove is_compiled flag in cpp tensor
GitOrigin-RevId: 15f90af735
|
4 years ago |
Megvii Engine Team
|
4f240ec2d3
|
refactor(mge/jit): make trace return any kind of output
GitOrigin-RevId: fd1265c661
|
4 years ago |
Megvii Engine Team
|
c6b552cf84
|
refactor(mge/functional): remove dependence to trace in functional implementations
GitOrigin-RevId: 0b18479fcc
|
4 years ago |
Megvii Engine Team
|
46d96478a1
|
feat(mge/tools): optimize statistical tools
BREAKING CHANGE:
GitOrigin-RevId: cd2a1acd11
|
4 years ago |
Megvii Engine Team
|
7f9d441cbb
|
docs(mge/module): add note about weight shape in conv
GitOrigin-RevId: 43e1f15968
|
4 years ago |
Megvii Engine Team
|
8be70d170c
|
docs(misc): correct docstring format broadly
GitOrigin-RevId: 45234ca07e
|
4 years ago |
Megvii Engine Team
|
53446bcea5
|
fix(imperative/dataset): fix VOC dataset
GitOrigin-RevId: a46a4c1173
|
4 years ago |
Megvii Engine Team
|
8e5bf94857
|
fix(mge/utils): fix bug of VarNode inplace operations
GitOrigin-RevId: fa9eec7079
|
4 years ago |
Megvii Engine Team
|
13b15fb08c
|
feat(megbrain): add correlation opr
GitOrigin-RevId: 6d44598891
|
4 years ago |
Megvii Engine Team
|
1997b1a289
|
feat(dnn/cuda): add correlation kernel
GitOrigin-RevId: 25e58b61e6
|
4 years ago |
Megvii Engine Team
|
6d376623de
|
fix(imperative/dataloader): collect garbage explicitly before starting a process in dataloader
GitOrigin-RevId: 3c3b51ad9c
|
4 years ago |
Megvii Engine Team
|
259894250b
|
feat(imperative/autodiff): add grad clip
GitOrigin-RevId: f344f4a233
|
4 years ago |
Megvii Engine Team
|
601a33a84d
|
refactor(mge/dtr): update dtr api
GitOrigin-RevId: dc366c65be
|
4 years ago |
Megvii Engine Team
|
c269a690e5
|
fix(mge/dtr): filter bad ops
GitOrigin-RevId: 380262dca0
|
4 years ago |
Megvii Engine Team
|
b230e14685
|
fix(mge/dump): fix dump_with_testcase_mge Varnode type mismatch
GitOrigin-RevId: 05618e5ac5
|
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
|
acb000d07f
|
fix(api_cache): fix serialization for conv_desc
GitOrigin-RevId: 95dbc9c685
|
4 years ago |
Megvii Engine Team
|
c4032222fa
|
fix(api_cache): lock api cache for thread safety
GitOrigin-RevId: 8a244677c3
|
4 years ago |
Megvii Engine Team
|
5419a95d1e
|
perf(cuda/conv): cache serval cudnn api
GitOrigin-RevId: 188c62cdd6
|
4 years ago |
Megvii Engine Team
|
19887942c8
|
feat(dnn/apicache): add generic apicache
GitOrigin-RevId: 40b8ac2ab6
|
4 years ago |
Megvii Engine Team
|
226f24fa8e
|
test(mge/distributed): add gather scatter reduce broadcast grad test
GitOrigin-RevId: 8245e11f1d
|
4 years ago |
Megvii Engine Team
|
aea5ac1363
|
fix(mge/distributed): fix gather scatter reduce broadcast autodiff
GitOrigin-RevId: 1c2250a079
|
4 years ago |
Megvii Engine Team
|
b1bf193e4d
|
feat(functional/loss): add reduction choices to loss functions
GitOrigin-RevId: a29e6bb4cf
|
4 years ago |