Caffe2移动端GPU支持列表

移动端是Caffe2是发力的方向之一。Caffe2支持通过OpenGL调用移动GPU。然而移动端GPU却无法保持桌面上对CPU的优势,多CPU核心配弱GPU更是安卓界的一大特色。即使强大如iPhone,对于iPhone 6s以下的设备,NNPACK加速的CPU实现也比Apple的MPSCNNConvolution性能好

Android

Android支持设备列表:

#if CAFFE2_ANDROID
// whitelist of supported GPUs
bool isSupportedRenderer() {
  static std::unordered_set<std::string> supported_renderers = {
      "Adreno (TM) 540",
      "Adreno (TM) 530",
      "Adreno (TM) 510",
      "Adreno (TM) 430",
      "Adreno (TM) 418",
      "Mali-G71",
      "Mali-T880",
      "NVIDIA Tegra"};
  std::string rendererStr((const char*)glGetString(GL_RENDERER));
  LOG(INFO) << "GL_RENDERER: " << rendererStr;

  int start = rendererStr.find_first_not_of(" ");
  int end = rendererStr.find_last_not_of(" ");
  rendererStr = rendererStr.substr(start, end - start + 1);
  return supported_renderers.count(rendererStr) > 0;
}
#endif

AndroidGLContext在构造函数中进行检查

Created with Raphaël 2.3.0 AndroidGLContext isSupportedDevice isSupportedRenderer

iOS

对于iOS,只支持iPhone 6以上

bool isSupportedDevice() {
#if CAFFE2_IOS
  return iPhoneVersion() >= 7; // iPhone 6 and up
#elif CAFFE2_ANDROID
  return isSupportedRenderer();
#else
  return false;
#endif
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值