Update encoder_gemm.h
This commit is contained in:
parent
793b92dca7
commit
4cbd73fcb9
|
@ -87,7 +87,7 @@ void generate_encoder_gemm_config(int batch_size,
|
||||||
cudaDataType_t CType;
|
cudaDataType_t CType;
|
||||||
cudaDataType_t computeType;
|
cudaDataType_t computeType;
|
||||||
int startAlgo, endAlgo;
|
int startAlgo, endAlgo;
|
||||||
const int ites = 1;
|
const int ites = 100;
|
||||||
struct timeval start, end;
|
struct timeval start, end;
|
||||||
|
|
||||||
if(sizeof(T) == sizeof(float)){
|
if(sizeof(T) == sizeof(float)){
|
||||||
|
|
Loading…
Reference in a new issue