Initiate checkCuda90Bug warning (#22757)

Guanheng Zhang · facebook-github-bot · commit f266a63eebf2 · 2019-07-12T09:55:09.000-07:00
Summary: Initiate checkCuda90Bug warning to THCudaBlas_Sgemm and THCudaBlas_Hgemm. #22034 Pull Request resolved: #22757 Differential Revision: D16223085 Pulled By: zhangguanheng66 fbshipit-source-id: 470c6cbaba16a3cec295993c2673f02008a602a6
diff --git a/aten/src/THC/THCBlas.cu b/aten/src/THC/THCBlas.cu
@@ -213,6 +213,7 @@ static void checkCuda90Bug(int i_m, int i_n, int i_k)
 /* Level 3 */
 void THCudaBlas_Sgemm(THCState *state, char transa, char transb, int64_t m, int64_t n, int64_t k, float alpha, float *a, int64_t lda, float *b, int64_t ldb, float beta, float *c, int64_t ldc)
 {
+  checkCuda90Bug((int)m, (int)n, (int)k);
   at::cuda::blas::gemm<float>(THCState_getCurrentStream(state), transa, transb, m, n, k, alpha, a, lda, b, ldb, beta, c, ldc);
 }
 
@@ -223,6 +224,7 @@ void THCudaBlas_Sgemm(THCState *state, char transa, char transb, int64_t m, int6
 
 void THCudaBlas_Hgemm(THCState *state, char transa, char transb, int64_t m, int64_t n, int64_t k, at::Half alpha, at::Half *a, int64_t lda, at::Half *b, int64_t ldb, at::Half beta, at::Half *c, int64_t ldc)
 {
+  checkCuda90Bug((int)m, (int)n, (int)k);
   at::cuda::blas::gemm<at::Half>(THCState_getCurrentStream(state), transa, transb, m, n, k, alpha, a, lda, b, ldb, beta, c, ldc);
 }
 

Original file line number	Diff line number	Diff line change
`@@ -213,6 +213,7 @@ static void checkCuda90Bug(int i_m, int i_n, int i_k)`
`213`	`213`	`/* Level 3 */`
`214`	`214`	`void THCudaBlas_Sgemm(THCState state, char transa, char transb, int64_t m, int64_t n, int64_t k, float alpha, float a, int64_t lda, float b, int64_t ldb, float beta, float c, int64_t ldc)`
`215`	`215`	`{`
	`216`	`+ checkCuda90Bug((int)m, (int)n, (int)k);`
`216`	`217`	`at::cuda::blas::gemm<float>(THCState_getCurrentStream(state), transa, transb, m, n, k, alpha, a, lda, b, ldb, beta, c, ldc);`
`217`	`218`	`}`
`218`	`219`
`@@ -223,6 +224,7 @@ void THCudaBlas_Sgemm(THCState *state, char transa, char transb, int64_t m, int6`
`223`	`224`
`224`	`225`	`void THCudaBlas_Hgemm(THCState state, char transa, char transb, int64_t m, int64_t n, int64_t k, at::Half alpha, at::Half a, int64_t lda, at::Half b, int64_t ldb, at::Half beta, at::Half c, int64_t ldc)`
`225`	`226`	`{`
	`227`	`+ checkCuda90Bug((int)m, (int)n, (int)k);`
`226`	`228`	`at::cuda::blas::gemm<at::Half>(THCState_getCurrentStream(state), transa, transb, m, n, k, alpha, a, lda, b, ldb, beta, c, ldc);`
`227`	`229`	`}`
`228`	`230`