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

fix(permute.h): incorrect comment in Tensor5DPermute20314 (#637)

Browse Source 
* fix(permute.h): incorrect comment in `Tensor5DPermute20314`

* typo in usage in example 39
This commit is contained in:
Wenzhuo Liu
2022-09-22 21:21:13 +08:00
committed by GitHub
parent 97bff52e8c
commit 7a458f00a6
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/cutlass/layout/permute.h
View File

@ -254,7 +254,7 @@ public:
};
/// Permute layout function for 5-D permuted tensors with output matrix (dimension as [M, N]) reshaped
/// as [M/T1, T1, T2, T3, N/T3]. Then perform permute([2, 0, 3, 1, 4]) on the corresponding output tensor.
/// as [M/T1, T1, T2, T3, N/T2/T3]. Then perform permute([2, 0, 3, 1, 4]) on the corresponding output tensor.
template <int T1, int T2, int T3>
class Tensor5DPermute20314 {
public: