Fix typo in docs, code comments (NVIDIA#429)

* [docs] fix typo in media/docs/layout.md * [docs] fix comment error * fix typo in include/cutlass/arch/simd_61.h * fix stride comment errors in TensorLayout
LeiWang1999 · Mar 16, 2022 · cd39c75 · cd39c75
1 parent b2e1e97
commit cd39c75
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/include/cutlass/arch/simd_sm61.h b/include/cutlass/arch/simd_sm61.h
@@ -23,7 +23,7 @@
  *
  **************************************************************************************************/
 /*! \file
-    \brief Templates exposing SIMD operators for SM60
+    \brief Templates exposing SIMD operators for SM61
 */
 
 #pragma once

diff --git a/include/cutlass/layout/tensor.h b/include/cutlass/layout/tensor.h
@@ -80,7 +80,7 @@ class TensorNHWC {
   // Data members
   //
 
-  /// Stride data member - [stride_w, stride_h, stride_n]
+  /// Stride data member - [c, wc, hwc]
   Stride stride_;
 
 public:
@@ -227,7 +227,7 @@ class TensorNCHW {
   // Data members
   //
 
-  /// Stride data member - [c, wc, hwc]
+  /// Stride data member - [w, hw, chw]
   Stride stride_;
 
 public:
@@ -312,7 +312,7 @@ class TensorNCxHWx {
   // Data members
   //
 
-  /// Stride data member - [c, wc, hwc]
+  /// Stride data member - [Interleave x w, Interleave x wh, hwc]
   Stride stride_;
 
 public:
@@ -421,7 +421,7 @@ class TensorCxRSKx {
   // Data members
   //
 
-  /// Stride data member - [c, wc, hwc]
+  /// Stride data member - [Interleave x n, Interleave x nw, Interleave x nwh]
   Stride stride_;
 
 public:

diff --git a/media/docs/layout.md b/media/docs/layout.md
@@ -126,8 +126,8 @@ This method is needed when an algorithm must define a buffer of arbitrary layout
 Example:
 ```c++
 
-typename ArbitraryLayout::TensorCord extent = make_Coord(...);
-typename ArbitraryLayout::TensorCord coord;
+typename ArbitraryLayout::TensorCoord extent = make_Coord(...);
+typename ArbitraryLayout::TensorCoord coord;
 
 ArbitraryLayout layout = ArbitraryLayout::packed(extent);