youngkingdom/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Typoes (#107)

Browse Source 
* Update splitk_gemm.cu

* Update gemm_bias_relu.cu

* Update mma_sm75.h
This commit is contained in:
hwu36
2020-07-13 17:25:52 -04:00
committed by GitHub
parent fd7e058d0c
commit 4dac7490e6
3 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/06_splitK_gemm/splitk_gemm.cu
View File

@ -205,7 +205,7 @@ int run() {
cutlass::HostTensor<ElementInputA, LayoutInputA> tensor_a(
problem_size.mk()); // <- Create matrix A with dimensions M x K
cutlass::HostTensor<ElementInputB, LayoutInputB> tensor_b(
problem_size.nk()); // <- Create matrix B with dimensions N x K
problem_size.kn()); // <- Create matrix B with dimensions K x N
cutlass::HostTensor<ElementOutput, LayoutOutput> tensor_c(
problem_size.mn()); // <- Create matrix C with dimensions M x N
cutlass::HostTensor<ElementOutput, LayoutOutput> tensor_d(