cudnnFindConvolutionBackwardFilterAlgorithmEx()
如何判断哪个算法最快?
主要是对卷积有很好的效果?
默认认为返回结果里第0个就是最好的
调用 try_all 的就是上层的conv逻辑,里面会在fwd or bwd时调用
一旦发现这个算法的参数之前测过在cache(hashmap)里,就直接用之前的结果:
否则去找最好的,用的cudnn的接口
本身上述代码并没有跟具体的算法强绑定。
其中 conv 上的cache 时判断的参数是这些 ConvolutionParams