Update encoder_gemm.h

This commit is contained in:
BO-YANG HSUEH 2020-03-18 19:22:39 +08:00 committed by GitHub
parent 793b92dca7
commit 4cbd73fcb9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -87,7 +87,7 @@ void generate_encoder_gemm_config(int batch_size,
cudaDataType_t CType;
cudaDataType_t computeType;
int startAlgo, endAlgo;
const int ites = 1;
const int ites = 100;
struct timeval start, end;
if(sizeof(T) == sizeof(float)){