From fcf40649feb6046fb9b1ed984fb9b19422cd5463 Mon Sep 17 00:00:00 2001
From: aamir <araihan@ece.ubc.ca>
Date: Sat, 21 Jul 2018 19:30:40 -0700
Subject: adding the cutlass examples

---
 cutlass-example/cutlass_example.cu | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)
 create mode 100644 cutlass-example/cutlass_example.cu

(limited to 'cutlass-example/cutlass_example.cu')

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);
+
+}
+
+
-- 
cgit v1.3