Megvii Engine Team
|
787a22a9d6
|
perf(tensor): implement __new__ in cpp
GitOrigin-RevId: 4defd249c3
|
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
|
21f5a7fcc0
|
fix(subgraph): fix device recognition and scalar propagate
GitOrigin-RevId: fd2fe8bec9
|
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
|
369c2ccc5a
|
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
|
3 years ago |
Megvii Engine Team
|
6082c353e7
|
feat(dnn/rocm): support bool in type_cvt and elemwise
GitOrigin-RevId: ad5ec7bc1c
|
4 years ago |
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 |