Megvii Engine Team
fe6299c52d
Revert "feat(mgb): allow output tensor's ptr change when record"
This reverts commit c610c8bf9a
.
GitOrigin-RevId: 2eb0504c5d
3 years ago
Megvii Engine Team
0a56a5b110
feat(cuda/comp_node): gets (maximum) GPU memory allocated/reserved
GitOrigin-RevId: da2cc22436
3 years ago
Megvii Engine Team
25ec2530ba
feat(whl/api/lar): enable megengine dll on Windows
1: reduce python whl package size
2: unify api link logic on all OS
3: add option: MGE_WINDOWS_BUILD_WITH_STATIC_CRT
--- default OFF
--- if build CRT(vc runtime) with STATIC with megengine.dll
some CRT api will crash, for example, flush, so if you
build with static megengine, and do not want to install CRT
you can set MGE_WINDOWS_BUILD_WITH_STATIC_CRT TRUE
--- how to install CRT:
https://docs.microsoft.com/en-us/cpp/windows/latest-supported-vc-redist?view=msvc-160
install VC_redist.x64.exe
4: rename megengine_export to megengine_shared(only export needed symbols ),
caused by runtime symbols conflict with pytorch
GitOrigin-RevId: 93d8d80f29
3 years ago
Megvii Engine Team
c17d603847
feat(mgb): allow output tensor's ptr change when record
GitOrigin-RevId: c610c8bf9a
3 years ago
Megvii Engine Team
d9a9d9d49e
fix(imperative/fastrun): set workspace limit for imperative rt
GitOrigin-RevId: 474dc691a3
3 years ago
Megvii Engine Team
369c2ccc5a
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
3 years ago
Megvii Engine Team
0a665ea452
feat(mge/device): enable to get cuda compute capability
GitOrigin-RevId: b5d3f2225c
3 years ago
Megvii Engine Team
1e3af4dd17
fix(mgb/comp_node): add more info in `comp_node.to_string()`
GitOrigin-RevId: 794a8847aa
3 years ago
Megvii Engine Team
3eb0505f9b
feat(imperative): add support for quantized conv transpose2d
GitOrigin-RevId: ffd6431299
3 years ago
Megvii Engine Team
8a73193c2d
feat(dtr): remove eviction threshold
GitOrigin-RevId: 35c2014bf3
3 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
1310ad4926
feat(dnn): add memory manager for enflame
GitOrigin-RevId: 72a810d959
4 years ago
Megvii Engine Team
9f2af2099c
feat(mgb): add enflame comp node
GitOrigin-RevId: 478c8538aa
4 years ago
Megvii Engine Team
a3ea1f153c
feat(mgb/opr): add fast profile and combined Execution strategy
GitOrigin-RevId: 843dc3a790
4 years ago
Megvii Engine Team
b6376c83a0
fix(mgb/comp_node): use extra physical device to decide whether reuse existed CompNodeImpl
GitOrigin-RevId: 5dddc68a84
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
4 years ago
Megvii Engine Team
cf53d9e0f8
fix(mgb/tensor): do tensor overlap check only when d2d and h2h
GitOrigin-RevId: 9125936a35
4 years ago
Megvii Engine Team
ab69883418
fix(mgb/core): use thread local for OpenCL recorder
GitOrigin-RevId: 76da0c2a24
4 years ago
Megvii Engine Team
8857456593
fix(mgb/core): use thread local fix multi thread use same compnode with recorder enabled
GitOrigin-RevId: 7d3daa866c
4 years ago
Megvii Engine Team
4a5e317063
fix(mgb/opr-mm): remove stream -4 for CollectiveComm
GitOrigin-RevId: 41ea88dfa1
4 years ago
Megvii Engine Team
a7b9ece4a7
feat(mgb/comp_node): add set_prealloc_config
GitOrigin-RevId: e725e7efdd
4 years ago
Megvii Engine Team
6f581906a1
refactor(mge/profiler): refactor profiler
GitOrigin-RevId: 279aa779a6
4 years ago
Megvii Engine Team
9510136223
fix(mgb/rocm): remove begin-internal of rocm
GitOrigin-RevId: 1523833fcb
4 years ago
Megvii Engine Team
00ef677249
fix(mgb): remove internal for cambricon and atlas
GitOrigin-RevId: 861e349eb4
4 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
8cf7150df0
feat(dnn/mge): add compnode multithread in python
GitOrigin-RevId: 47373d291d
5 years ago
Megvii Engine Team
f91881ffdc
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
5 years ago