CUTLASS 2.10 updates (#622)

Co-authored-by: Aniket Shivam <ashivam@nvidia.com>
2022-09-12 18:26:30 -07:00
parent beae168f90
commit e773429f7e
96 changed files with 8365 additions and 1667 deletions
--- a/tools/library/scripts/pycutlass/docs/source/cutlass.rst
+++ b/tools/library/scripts/pycutlass/docs/source/cutlass.rst
@ -1,2 +1,100 @@
 cutlass
 =======
+
+.. rubric:: Operator Classification
+
+.. autoclass:: cutlass.OpClass
+    :members:
+
+.. rubric:: GEMM Layout
+
+.. autoclass:: cutlass.RowMajor
+    :members:
+
+.. autoclass:: cutlass.ColumnMajor
+    :members:
+
+.. autoclass:: cutlass.RowMajorInterleaved32
+    :members:
+
+.. autoclass:: cutlass.ColumnMajorInterleaved32
+    :members:
+
+.. rubric:: Conv Layout
+
+.. autoclass:: cutlass.TensorNHWC
+    :members:
+
+.. autoclass:: cutlass.TensorNC32HW32
+    :members:
+
+.. autoclass:: cutlass.TensorC32RSK32
+    :members:
+
+.. rubric:: Threadblock Swizzle
+
+.. autoclass:: cutlass.dim3
+    :special-members:
+    :members:
+
+.. autoclass:: cutlass.IdentitySwizzle1
+    :special-members:
+    :members:
+
+.. autoclass:: cutlass.IdentitySwizzle2
+    :special-members:
+    :members:
+
+.. autoclass:: cutlass.IdentitySwizzle4
+    :special-members:
+    :members:
+
+.. autoclass:: cutlass.IdentitySwizzle8
+    :special-members:
+    :members:
+
+.. autoclass:: cutlass.HorizontalSwizzle
+    :special-members:
+    :members:
+
+.. autoclass:: cutlass.BatchedIdentitySwizzle
+    :special-members:
+    :members:
+
+.. autoclass:: cutlass.StridedDgradIdentitySwizzle1
+    :special-members:
+    :members:
+
+.. autoclass:: cutlass.StridedDgradIdentitySwizzle4
+    :special-members:
+    :members:
+
+.. autoclass:: cutlass.StridedDgradHorizontalSwizzle
+    :special-members:
+    :members:
+
+.. rubric:: Coordinates
+
+.. autoclass:: cutlass.Tensor4DCoord
+    :special-members:
+    :members:
+
+.. autoclass:: cutlass.Tensor3DCoord
+    :special-members:
+    :members:
+
+.. autoclass:: cutlass.MatrixCoord
+    :special-members:
+    :members:
+
+
+.. rubric:: Convolution
+
+.. autoclass:: cutlass.conv.Operator
+    :members:
+
+.. autoclass:: cutlass.conv.IteratorAlgorithm
+    :members:
+
+.. autoclass:: cutlass.conv.StrideSupport
+    :members: