machine-learning - CAFFE中某层的.cpp和.cu的关系

Question

我发现一个层可以在CPU(.cpp)或GPU(.cu)版本中实现。例如，批量归一化层有两个源文件:batch_norm_layer.cpp 和 batch_norm_layer.cu。

当我编译代码时，我通过取消注释 USE_CUDNN := 1 来选择 CUDA 模式。然后，当我使用批量归一化层时，CAFFE 将执行 .cu 文件中的所有操作，而不从 .cpp 文件中获取任何信息。我说得对吗？

我问这个问题是因为我不确定我的CPU实现是否正确，但我确定GPU实现是正确的。因此，我不知道 CPU 代码中的任何错误是否会影响我的 GPU 代码。谢谢

Answer 1

LayerSetup 是在 .cpp 代码中执行的，因此您需要正确执行此操作。否则，你是对的(假设你选择了 solver_mode: GPU)，你的 .cu 代码应该运行前向和后向传递。