Compile fix

6 years ago · f7cb538b32
parent 584f840b40
commit f7cb538b32
1 changed files with 2 additions and 0 deletions
--- 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 << <num_blocks, BLOCK, 0, get_cuda_stream() >> > (
            M, N, K,