Fixed bug in Tensor Cores V100 (1. Desc in Batch norm, 2. Manually selected...

AlexeyAB

2018-04-14 eb9c88ef734d693e65ec35036811363a35e6b5d3

 src/gemm.h

@@ -1,6 +1,11 @@
#ifndef GEMM_H
#define GEMM_H

void gemm_bin(int M, int N, int K, float ALPHA, 
        char  *A, int lda, 
        float *B, int ldb,
        float *C, int ldc);
        
void gemm(int TA, int TB, int M, int N, int K, float ALPHA, 
                    float *A, int lda, 
                    float *B, int ldb,