Megvii Engine Team
03ab8136e7
fix(core): fix asan error cause by wild thread_pool ptr
GitOrigin-RevId: b1c1b452cd
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
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
1bec737d1a
feat(distributed): support distributed opr for rocm
GitOrigin-RevId: 4840100d07
5 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
9f2af2099c
feat(mgb): add enflame comp node
GitOrigin-RevId: 478c8538aa
5 years ago
Megvii Engine Team
e1c83d8d51
fix(mgb/core): add warning information about const_var_shape when record mode
GitOrigin-RevId: a99f9c4e5d
4 years ago
Megvii Engine Team
fc8b501bb8
refactor(mgb/core): refactor cpu compnode so that default cpu has no ability to record
GitOrigin-RevId: 7de4771476
4 years ago
Megvii Engine Team
47bdb4ef1d
fix(mgb/core): add MGB_HAVE_THREAD macro to thread_local
GitOrigin-RevId: cfcb50b041
5 years ago
Megvii Engine Team
b6376c83a0
fix(mgb/comp_node): use extra physical device to decide whether reuse existed CompNodeImpl
GitOrigin-RevId: 5dddc68a84
5 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
5 years ago
Megvii Engine Team
62637fc46e
refactor(mgb/utils): allow user set different spin time for ASyncSCQueue
GitOrigin-RevId: 1c6c4a7a16
5 years ago
Megvii Engine Team
a5609f3bbf
fix(cambricon): fix cross cn copy for cambricon
GitOrigin-RevId: 21942a82a3
5 years ago
Megvii Engine Team
15dd5e1a0f
fix(mgb/core): fix memory management release cambricon var issue
GitOrigin-RevId: abf881978c
5 years ago
Megvii Engine Team
cf53d9e0f8
fix(mgb/tensor): do tensor overlap check only when d2d and h2h
GitOrigin-RevId: 9125936a35
5 years ago
Megvii Engine Team
4f527af4ae
chore(mbg/core): fix the repeat code
GitOrigin-RevId: 3d4c91a2c9
5 years ago
Megvii Engine Team
005ead5aea
docs(cpu/comp_node): note cpu_default()
GitOrigin-RevId: 3e3dbced65
5 years ago
Megvii Engine Team
3635af6274
style(atlas): add comment for async d2d
GitOrigin-RevId: 606a56ac4e
5 years ago
Megvii Engine Team
d68d4d1d99
perf(atlas): use async d2d
GitOrigin-RevId: 55914631cb
5 years ago
Megvii Engine Team
cb8e53635a
fix(mgb/core): disable thread_local in ios due to x-code bug
GitOrigin-RevId: 4a7c838810
5 years ago
Megvii Engine Team
ab69883418
fix(mgb/core): use thread local for OpenCL recorder
GitOrigin-RevId: 76da0c2a24
5 years ago
Megvii Engine Team
8857456593
fix(mgb/core): use thread local fix multi thread use same compnode with recorder enabled
GitOrigin-RevId: 7d3daa866c
5 years ago
Megvii Engine Team
10106341fe
fix(mgb/opr): fix take CpuDispatchableBase::EventImpl as CpuEventImpl
GitOrigin-RevId: 07aa850837
5 years ago
Megvii Engine Team
b8ddca4c38
fix(atlas): add MGB_USE_ATLAS_ASYNC_API to enable async api
GitOrigin-RevId: ab821f4966
5 years ago
Megvii Engine Team
a7b9ece4a7
feat(mgb/comp_node): add set_prealloc_config
GitOrigin-RevId: e725e7efdd
5 years ago
Megvii Engine Team
066da0bff1
fix(mgb/comp_node): add lock_guard to sd.mtx & update warning message & add test
GitOrigin-RevId: ae0bf4b479
5 years ago
Megvii Engine Team
6f581906a1
refactor(mge/profiler): refactor profiler
GitOrigin-RevId: 279aa779a6
5 years ago
Megvii Engine Team
9510136223
fix(mgb/rocm): remove begin-internal of rocm
GitOrigin-RevId: 1523833fcb
5 years ago
Megvii Engine Team
00ef677249
fix(mgb): remove internal for cambricon and atlas
GitOrigin-RevId: 861e349eb4
5 years ago
Megvii Engine Team
e8571cca51
fix(mgb/cuda): fix cuda host alloc set device
GitOrigin-RevId: f4756e8981
5 years ago
Megvii Engine Team
09b5f3d434
fix(mgb/core): fix multi thread pool deactive and multi thread conflict
GitOrigin-RevId: 36787a08a5
5 years ago
Megvii Engine Team
734c498d27
perf(mgb/core): improve DevMemAlloc when it has single stream
GitOrigin-RevId: 61874faa6d
5 years ago
Megvii Engine Team
b43fb1a97c
perf(mgb): add CUDA host memory allocator
test(mgb): add SimpleCachingAlloc test
GitOrigin-RevId: 17f381e4ac
5 years ago
Megvii Engine Team
9e9e8ca0fe
fix(mgb): fix some compile errors
GitOrigin-RevId: ec38f8e15a
5 years ago
Megvii Engine Team
318f6d75fa
fix(mge): fix some warnings
GitOrigin-RevId: 38b285f991
5 years ago
Megvii Engine Team
f4ead78852
feat(mgb): static allocation with given padding
GitOrigin-RevId: fdf2de8ad6
5 years ago
Megvii Engine Team
ff308e3b62
feat(mgb/comp_node): generate uid for cuda comp node
GitOrigin-RevId: 34fa5a2fb6
5 years ago
Megvii Engine Team
1af4863179
fix(dnn/fallback): fix conv1x1's is_preferred method
GitOrigin-RevId: 6d34080d25
5 years ago
zjd1988
d34e385dd7
assert locator.device < sd.MAX_NR_DEVICE when bool(avaiable_node) = true
Signed-off-by: zjd1988 <942002795@qq.com>
5 years ago
Megvii Engine Team
8cf7150df0
feat(dnn/mge): add compnode multithread in python
GitOrigin-RevId: 47373d291d
5 years ago
Megvii Engine Team
856ef62721
feat(mgb/core): support copy DeviceTensorND from cpu to cuda
GitOrigin-RevId: d56f4ebf1f
5 years ago
Megvii Engine Team
f91881ffdc
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
6 years ago