CUTLASS 3.2.1 (#1113)

* Updates for 3.2.1 release.

* Minor fix in gemm op profiler for raster order.

* Add scheduler mapping for raster order in the kernels.

tools/library/include/cutlass/library/library.h

@@ -291,6 +291,7 @@ struct GemmUniversalArguments {
   // Needed for some 3.x kernels
   int sm_count;
 
+  library::RasterOrder raster_order;
 };
 
 /////////////////////////////////////////////////////////////////////////////////////////////////

tools/library/include/cutlass/library/types.h

@@ -250,6 +250,13 @@ enum class EpilogueKind {
   kInvalid
 };
 
+enum class RasterOrder {
+  kAlongN,
+  kAlongM,
+  kHeuristic,
+  kInvalid
+};
+
 /////////////////////////////////////////////////////////////////////////////////////////////////
 
 } // namespace library

tools/library/include/cutlass/library/util.h

@@ -170,6 +170,13 @@ char const *to_string(ConvKind type, bool pretty = false);
 template <>
 ConvKind from_string<ConvKind>(std::string const &str);
 
+/// Converts a RasterOrder enumerant to a string
+char const *to_string(RasterOrder type, bool pretty = false);
+
+/// Convers a RasterOrder enumerant from a string
+template<>
+RasterOrder from_string<RasterOrder>(std::string const &str);
+
 /// Lexical cast from int64_t to string
 std::string lexical_cast(int64_t int_value);