Megvii Engine Team
|
0a665ea452
|
feat(mge/device): enable to get cuda compute capability
GitOrigin-RevId: b5d3f2225c
|
3 years ago |
Megvii Engine Team
|
8c47c1f149
|
perf(syncbn): reimplement with subgraph
GitOrigin-RevId: 13e7e3d3c0
|
3 years ago |
Megvii Engine Team
|
a605f38b26
|
refactor(opmeth): add OpMethCache struct
GitOrigin-RevId: c1ebe15672
|
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
|
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
|
87f4b46e3f
|
perf(mge/imperative): move convert_inputs from python to C++
GitOrigin-RevId: baef3d348c
|
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
|
2e9ba679b5
|
feat(mge/device): __repr__ method will show physical device
GitOrigin-RevId: 050c3864a7
|
4 years ago |
Megvii Engine Team
|
a7b9ece4a7
|
feat(mgb/comp_node): add set_prealloc_config
GitOrigin-RevId: e725e7efdd
|
4 years ago |
Megvii Engine Team
|
2cc85487d1
|
fix(imperative): fix hardcode of default device
GitOrigin-RevId: 722c4debfa
|
4 years ago |
Megvii Engine Team
|
d4bad71190
|
feat(mge): add jit.trace
GitOrigin-RevId: ec647324c0
|
4 years ago |
Megvii Engine Team
|
c5e433b568
|
refactor(mge): cpp license
GitOrigin-RevId: 2f67f84724
|
4 years ago |
Megvii Engine Team
|
3840c1f433
|
fix(mge/imperative): use logical reprentation of compnode
GitOrigin-RevId: 9d104de3aa
|
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 |