Megvii Engine Team
2f68aeb9b6
feat(imperative/jit): let trace support empty IO
GitOrigin-RevId: 97a55242bf
4 years ago
Megvii Engine Team
809d5056cd
feat(mge/distributed): enable pt shm allreduce
GitOrigin-RevId: 1dd5a02a51
4 years ago
Megvii Engine Team
88898e63a5
fix(mgb): replace if_constexpr with runtime function to avoid potential
bug
GitOrigin-RevId: 27fe093d50
4 years ago
Megvii Engine Team
1cfdbc565c
feat(dnn): add deterministic max pooling
GitOrigin-RevId: 9ab4c7a748
4 years ago
Megvii Engine Team
933dd9a497
feat(mge/distributed): add cuda env check before forked thread
style(core/comp_node): reformat code
GitOrigin-RevId: 372452a8eb
4 years ago
Megvii Engine Team
2a54196117
fix(tee): fix tee link
GitOrigin-RevId: db7b98524d
4 years ago
Megvii Engine Team
a5060a2bfe
feat(mgb/opr): add check_has_inf kernel and opr
GitOrigin-RevId: 0d042dbfce
4 years ago
Megvii Engine Team
3597a6dbd7
feat(dnn/arm): nchw_nchw44 conv support 1x1s1
GitOrigin-RevId: 8c8f7d7c76
4 years ago
Megvii Engine Team
40085acbae
fix(mgb): remove unnecessary cudnn8 warning
GitOrigin-RevId: 04cf1bfca9
4 years ago
Megvii Engine Team
54a4d70eb5
feat(src/serialization): add support of serializing metadata
GitOrigin-RevId: b563c94451
4 years ago
Megvii Engine Team
721091faf0
fix(core): fix thread local is not supported in ios
GitOrigin-RevId: b7a6928f0b
4 years ago
Megvii Engine Team
62bd6c823b
feat(cmake/debug): misc for build
* add asan build option
* fix cpuinfo build opt level
* fix host release build with out debug info
* opt "fix lite bazel/cmake symbols MR"
* other misc build opt
GitOrigin-RevId: 6ca286e195
4 years ago
Megvii Engine Team
3e4e4c4604
feat(mgb/jit): add graph_opt_config and jit_config interfaces
GitOrigin-RevId: 170d9eeab2
4 years ago
Megvii Engine Team
1c7d0802ab
fix(cuda): remove cuda driver version check and runtime minor version
GitOrigin-RevId: 4463beccf1
4 years ago
tpoisonooo
7038a7f5d0
fix(quant): fix spell error
4 years ago
Megvii Engine Team
355153e158
feat(mge/dtr): add DTR in computing graph
GitOrigin-RevId: 8941810319
4 years ago
Megvii Engine Team
76f4f97536
refactor(sublinear): add SeqModifierBase
GitOrigin-RevId: 2d0393be6b
4 years ago
Megvii Engine Team
f584416aa2
fix(dnn/bn): revise the conditions for inplace flag
GitOrigin-RevId: 59a104bf6a
4 years ago
Megvii Engine Team
2eea00097c
feat(mgb): add fast run batch size graph option
GitOrigin-RevId: 94e333ec80
4 years ago
Megvii Engine Team
47dcdf3e17
fix(mgb/core): fix dtype and resize modifiers for tensor
GitOrigin-RevId: a9d95a4cd8
4 years ago
Megvii Engine Team
29f7cdb84a
fix(mgb/opr): correct nvof out shape computation
GitOrigin-RevId: 16bf086e92
4 years ago
Megvii Engine Team
03ab8136e7
fix(core): fix asan error cause by wild thread_pool ptr
GitOrigin-RevId: b1c1b452cd
4 years ago
Megvii Engine Team
0fb9cc41e4
fix(gopt): fix nchw64 opt pass
GitOrigin-RevId: dec18d1ab1
4 years ago
Megvii Engine Team
86b69cacd0
fix(dnn): fixes for int4
GitOrigin-RevId: 845e164fd3
4 years ago
Megvii Engine Team
adf75a291d
perf(dnn/cuda): add sass int4 128x128
GitOrigin-RevId: 1bc5482102
4 years ago
Megvii Engine Team
8da2f698a3
feat(dnn/cuda): support warp perspective/pooling op when channel not aligned to 64
GitOrigin-RevId: 39f29ec990
4 years ago
Megvii Engine Team
c218d4b029
feat(dnn/cuda): fallback conv qs4 support channel not aligend to 64
GitOrigin-RevId: f0d080f35c
4 years ago
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