CUTLASS 3.1 Python interface documentation (#917)

* Add 12.1 Dockerfile * Add 3.1 docs
2023-04-18 15:11:35 -04:00
parent 54bebe417d
commit 9a83bd3381
81 changed files with 18997 additions and 10 deletions
@@ -0,0 +1,26 @@
+Operations
+==========
+
+GEMM
+----
+
+.. automodule:: cutlass.op.gemm
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+Grouped GEMM
+------------
+
+.. automodule:: cutlass.op.gemm_grouped
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+Operation
+---------
+
+.. automodule:: cutlass.op.op
+   :members:
+   :undoc-members:
+   :show-inheritance: