OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
2853 Commits
28 Branches
25 MB
4628 Download
Tree: dbd9483993
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'dbd9483993'
${ noResults }
MegEngine/dnn/src/common/cpuinfo_arch_vendor.h

cpuinfo_arch_vendor.h 318 B
Raw Normal View History

feat(dnn/arm): add cpuinfo runtime check for x86 and arm GitOrigin-RevId: c2020a344ef286c6104d78525e52eb4a1e86ab82
4 years ago
feat(dnn/arm): add fp32 asm gemm for a53 a55 and i8i8i16 gemm for a72 a53 GitOrigin-RevId: a049c33f2bf1e737630de263161d7e32be2ba645
4 years ago
​
feat(dnn/arm): add cpuinfo runtime check for x86 and arm GitOrigin-RevId: c2020a344ef286c6104d78525e52eb4a1e86ab82
4 years ago
​
​
​
​
​
​
​
​
​
​
​
 123456789101112131415 
  1. #pragma once

  2. #include "src/common/utils.h"

  3. #if defined(MGB_ENABLE_CPUINFO_CHECK) && MGB_ENABLE_CPUINFO

  4. 

  5. #include <cpuinfo.h>

  6. 

  7. namespace megdnn {

  8. 

  9. const char* vendor_to_string(enum cpuinfo_vendor vendor);

  10. const char* uarch_to_string(enum cpuinfo_uarch uarch);

  11. 

  12. }  // namespace megdnn

  13. #endif

  14. 

  15. // vim: syntax=cpp.doxygen

MegEngine 安装包中集成了使用 GPU 运行代码所需的 CUDA 环境，不用区分 CPU 和 GPU 版。 如果想要运行 GPU 程序，请确保机器本身配有 GPU 硬件设备并安装好驱动。 如果你想体验在云端 GPU 算力平台进行深度学习开发的感觉，欢迎访问 MegStudio 平台