checking sgemm and dgemm interface

Author: AamirRaihan

cutlass.cu

@@ -2,12 +2,16 @@
 #include <cutlass/wmma_matrix.h>
 #include <cutlass/gemm/gemm.h>
 #include <cutlass/gemm/wmma_gemm_traits.h>
+#include "cutlass/gemm/sgemm_traits.h"
+#include "cutlass/gemm/dgemm_traits.h"
 #include <gemm-test/gemm_testbed.h>
 #include <gemm-test/gemm.h>
 
 int main(int argc, char* argv[]) {
 
 	#include <gemm-test/wmma_tests.h>
+	#include <gemm-test/sgemm_tests.h>
+	#include <gemm-test/dgemm_tests.h>
 }
 
 

gemm-test/dgemm_tests.h

@@ -0,0 +1,19 @@
+  typedef cutlass::gemm::DgemmTraits<cutlass::MatrixLayout::kColumnMajor,
+                                     cutlass::MatrixLayout::kRowMajor,
+                                     cutlass::Shape<8, 32, 64> > DGemmTraits1;
+  run_gemm<DGemmTraits1>(64, 32, 8);
+
+
+
+  typedef cutlass::gemm::DgemmTraits<cutlass::MatrixLayout::kColumnMajor,
+                                     cutlass::MatrixLayout::kRowMajor,
+                                     cutlass::Shape<8, 32, 64> > DGemmTraits2;
+  run_gemm<DGemmTraits2>(256, 128, 64);
+
+
+
+  typedef cutlass::gemm::DgemmTraits<cutlass::MatrixLayout::kColumnMajor,
+                                     cutlass::MatrixLayout::kRowMajor,
+                                     cutlass::Shape<8, 64, 64> > DGemmTraits3;
+  run_gemm<DGemmTraits3>(64, 64, 8);
+

gemm-test/sgemm_tests.h

@@ -0,0 +1,16 @@
+  typedef cutlass::gemm::SgemmTraits<cutlass::MatrixLayout::kColumnMajor,
+    cutlass::MatrixLayout::kRowMajor, cutlass::Shape<8, 128, 128> >
+    SgemmTraits1;
+  run_gemm<SgemmTraits1>(1024, 512, 8);
+
+
+  typedef cutlass::gemm::SgemmTraits<cutlass::MatrixLayout::kColumnMajor,
+                                     cutlass::MatrixLayout::kRowMajor, cutlass::Shape<8, 128, 128> >
+      SgemmTraits2;
+  run_gemm<SgemmTraits2>(128, 81, 1);
+
+
+  typedef cutlass::gemm::SgemmTraits<cutlass::MatrixLayout::kColumnMajor,
+                                     cutlass::MatrixLayout::kRowMajor, cutlass::Shape<8, 128, 128> >
+      SgemmTraits3;
+  run_gemm<SgemmTraits3>(128, 112, 8);