mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
14 lines
419 B
C++
14 lines
419 B
C++
#include "caffe2/operators/conv_op_cache_cudnn.h"
|
|
|
|
#include <cudnn.h>
|
|
|
|
#include "caffe2/core/logging.h"
|
|
#include "caffe2/core/tensor.h"
|
|
|
|
namespace caffe2 {
|
|
|
|
template class AlgorithmsCache<cudnnConvolutionFwdAlgo_t>;
|
|
template class AlgorithmsCache<cudnnConvolutionBwdFilterAlgo_t>;
|
|
template class AlgorithmsCache<cudnnConvolutionBwdDataAlgo_t>;
|
|
template class AlgorithmsCache<int>; // For testing.
|
|
} // namespace caffe2
|