Gpu版本

[zhanglaplace]

你好，请问你实现了gpu版本的么

[zhanglonghao1992]

@zhanglaplace 我最近在试着写gpu版本但是运行报错有点问题，请问你实现了吗

[zhanglaplace]

@zhanglonghao1992 我这边也是，在CUDA LOOP的时候调用map或者weighted的时候也报错了，但是cpu版本速度没啥差别感觉。

[zhanglonghao1992]

@zhanglaplace global前缀的函数里像vector map啥的操作都不能用，换成array是可以的。但是我报错是因为其他东西。。请问能加个q交流下吗？819375556这是我的

[zhanglonghao1992]

@zhanglaplace 想问下怎么让这层执行forward cpu，不调用gpu

[zhanglaplace]

@zhanglonghao1992 没有gpu实现的时候自动执行CPU的。。。

[zhanglonghao1992]

@zhanglaplace 那我把cu文件删了就可以了？。。。。。。。。。。。

[zhanglaplace]

@zhanglonghao1992 Caffe Layer定义的前向为虚函数，基类的里面gpu里面调用cpu，所以你在头文件注释掉gpu的实现就可以了