From f7cb538b32d95e99a6751bf312f7c846419389d7 Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Sun, 3 Feb 2019 00:37:00 +0300 Subject: [PATCH] Compile fix --- src/im2col_kernels.cu | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/im2col_kernels.cu b/src/im2col_kernels.cu index 70fd2310..7b347857 100644 --- a/src/im2col_kernels.cu +++ b/src/im2col_kernels.cu @@ -1779,6 +1779,7 @@ void gemm_nn_custom_bin_mean_transposed_gpu(int M, int N, int K, //if (M % 8 == 0 && N % 8 == 0 && M == 128) //if (M >= 32) // l.n >= 32 +#if CUDA_VERSION >= 10000 if (1) { const int M_aligned = M + (8 - (M % 8)); @@ -1799,6 +1800,7 @@ void gemm_nn_custom_bin_mean_transposed_gpu(int M, int N, int K, //getchar(); } else +#endif //# CUDA_VERSION >= 10000 { gemm_nn_custom_bin_mean_transposed_gpu_kernel << > > ( M, N, K,