diff --git a/media/docs/profiler.md b/media/docs/profiler.md index 35106f26..4cf71649 100644 --- a/media/docs/profiler.md +++ b/media/docs/profiler.md @@ -210,8 +210,8 @@ GEMM [int] --inst_k,--instruction-shape::k Math instruction shape in the K dimension [int] --min_cc,--minimum-compute-capability Minimum device compute capability [int] --max_cc,--maximum-compute-capability Maximum device compute capability - [enum] --raster_order={H|M|N} If supported by kernel, sets the tile raster direction - [int] --swizzle_size If supported by kernel, sets the 2D tile swizzle extent + [enum] --raster_order={heuristic|H|along_m|M|along_n|N} If supported by kernel, sets the tile raster direction + [int] --swizzle_size={1,2,4,8} If supported by kernel, sets the 2D tile swizzle extent (In Hopper, other values will be rounded down to the nearest supported value) Examples: Profile a particular problem size: diff --git a/tools/library/src/util.cu b/tools/library/src/util.cu index 927806d2..314ec6c0 100644 --- a/tools/library/src/util.cu +++ b/tools/library/src/util.cu @@ -1010,12 +1010,13 @@ ConvKind from_string(std::string const &str) { static struct { char const *text; char const *pretty; + char const *character; RasterOrder enumerant; } RasterOrder_enumerants[] = { - {"along_n", "", RasterOrder::kAlongN}, - {"along_m", "", RasterOrder::kAlongM}, - {"heuristic", "", RasterOrder::kHeuristic}, + {"along_n", "", "N", RasterOrder::kAlongN}, + {"along_m", "", "M", RasterOrder::kAlongM}, + {"heuristic", "", "H", RasterOrder::kHeuristic}, }; /// Converts a RasterOrder enumerant to a string @@ -1042,7 +1043,8 @@ RasterOrder from_string(std::string const &str) { for (auto const & possible : RasterOrder_enumerants) { if ((str.compare(possible.text) == 0) || - (str.compare(possible.pretty) == 0)) { + (str.compare(possible.pretty) == 0) || + (str.compare(possible.character) == 0)) { return possible.enumerant; } }