Megvii Engine Team
d60d028a40
feat(mge/device): enable to get cuda/cudnn/tensorrt version
GitOrigin-RevId: 5864c61d10
3 years ago
Megvii Engine Team
c2e9860feb
chore(license): remove all license in file header
GitOrigin-RevId: a0e31247a6
3 years ago
Megvii Engine Team
1c2a323e78
feat(mge): add warning message when mismatched cuda sm is detected
GitOrigin-RevId: f78c79eb06
3 years ago
Megvii Engine Team
877bda4180
perf(mge): improve cross stream memory borrowing
GitOrigin-RevId: c68977c5dc
3 years ago
Megvii Engine Team
17f2dffb5b
fix(imperative/cpu/multithread): fix multithread at imperative
GitOrigin-RevId: 9120d5cb48
3 years ago
Megvii Engine Team
7ea104d788
Revert "fix(mge): replace _full_sync by sync"
This reverts commit e36ef45464
GitOrigin-RevId: 2d913c8ac9
3 years ago
Megvii Engine Team
8b60bdfa10
fix(mge): replace _full_sync by sync
GitOrigin-RevId: e36ef45464
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
fba523a196
fix(mge/device): fix the directive content block
GitOrigin-RevId: 539feb7956
3 years ago
Megvii Engine Team
182ca25d45
feat(mge/imperative): add coalesce_all_free_memory mge api
GitOrigin-RevId: ea05b1e9a1
3 years ago
Megvii Engine Team
4de62ad6cd
docs(mge/device): fix docstring of get_cuda_compute_capability
GitOrigin-RevId: 4ec04e8b9b
3 years ago
Megvii Engine Team
dff7719e87
feat(mge/distributed): add preload host data with op
fix(mge/distributed): change api name with preload
fix(mge/distributed): fix recursive model in preload tensor
fix(mge/distributed): fix recursive when cache contain None
GitOrigin-RevId: 80e2a6dd70
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
c5f8b58385
docs(docstring): transfer to google style
GitOrigin-RevId: a71245c553
3 years ago
Megvii Engine Team
5c131d2a74
fix(device): fix get_device_count for xpu
GitOrigin-RevId: 6c77f507f3
4 years ago
Megvii Engine Team
dee5a10af9
feat(distributed): auto detect device and backend when init group
GitOrigin-RevId: 90be2d5b4d
4 years ago
Megvii Engine Team
c8697a7005
feat(imperative/src): python wrapper for cambricon and atlas runtime opr
GitOrigin-RevId: bd969d1339
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
4 years ago
Megvii Engine Team
ab82c8da79
feat(mge/device): add python binding for get_mem_status_bytes
GitOrigin-RevId: aa0d3264d6
4 years ago
Megvii Engine Team
b309890c66
docs(mge): pytest for sphinx docstring
GitOrigin-RevId: 8bed12562a
4 years ago
Megvii Engine Team
d225cbcdbe
feat(mge): add device name check
GitOrigin-RevId: d9910b6275
4 years ago
Megvii Engine Team
026af62042
docs(mge): docs typo fix
GitOrigin-RevId: 851f6de02f
4 years ago
Megvii Engine Team
a7b9ece4a7
feat(mgb/comp_node): add set_prealloc_config
GitOrigin-RevId: e725e7efdd
4 years ago
Megvii Engine Team
066da0bff1
fix(mgb/comp_node): add lock_guard to sd.mtx & update warning message & add test
GitOrigin-RevId: ae0bf4b479
4 years ago
Megvii Engine Team
44d0b5daf5
feat(imperative): enable to() to copy to device
GitOrigin-RevId: f9caf17d24
4 years ago
Megvii Engine Team
2cc85487d1
fix(imperative): fix hardcode of default device
GitOrigin-RevId: 722c4debfa
4 years ago
Megvii Engine Team
6b380e8965
feat(mge/imperative): run oss test and restore cmake list build items
GitOrigin-RevId: 11411b6964
4 years ago