youngkingdom/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Profiler docs and argument update for raster order (#1667)

Browse Source
This commit is contained in:
dePaul Miller
2024-07-31 13:40:10 -07:00
committed by GitHub
parent fbd116c0e5
commit 8b2a0408bd
2 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tools/library/src/util.cu
View File

@ -1010,12 +1010,13 @@ ConvKind from_string<ConvKind>(std::string const &str) {
static struct {
char const *text;
char const *pretty;
char const *character;
RasterOrder enumerant;
}
RasterOrder_enumerants[] = {
{"along_n", "<along_n>", RasterOrder::kAlongN},
{"along_m", "<along_m>", RasterOrder::kAlongM},
{"heuristic", "<heuristic>", RasterOrder::kHeuristic},
{"along_n", "<along_n>", "N", RasterOrder::kAlongN},
{"along_m", "<along_m>", "M", RasterOrder::kAlongM},
{"heuristic", "<heuristic>", "H", RasterOrder::kHeuristic},
};
/// Converts a RasterOrder enumerant to a string
@ -1042,7 +1043,8 @@ RasterOrder from_string<RasterOrder>(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;
}
}