diff options
Diffstat (limited to 'cutlass-example/cutlass_example.cu')
| -rw-r--r-- | cutlass-example/cutlass_example.cu | 17 |
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); + +} + + |