adding the cutlass examples

1 files changed, 17 insertions, 0 deletions

diff --git a/cutlass-example/cutlass_example.cu b/cutlass-example/cutlass_example.cu
new file mode 100644
index 0000000..d3e8f89
--- /dev/null
+++ b/cutlass-example/cutlass_example.cu
@@ -0,0 +1,17 @@
+//added by me
+#include <cutlass/wmma_matrix.h>
+#include <cutlass/gemm/gemm.h>
+#include <cutlass/gemm/wmma_gemm_traits.h>
+#include <gemm_testbed.h>
+#include <gemm.h>
+
+int main(int argc, char* argv[]) {
+    typedef cutlass::gemm::WmmaGemmTraits<cutlass::MatrixLayout::kColumnMajor,
+                                        cutlass::MatrixLayout::kRowMajor, 
+                                        cutlass::Shape<32, 128, 128> >
+      WmmaGemmTraits;
+  run_gemm<WmmaGemmTraits>(256, 256, 128);
+
+}
+
+