Add description for profiler operations

profiler/src/profile_softmax.cpp

@@ -166,4 +166,4 @@ int profile_softmax(int argc, char* argv[])
 //     return 0;
 // }
 
-REGISTER_PROFILER_OPERATION("softmax", profile_softmax)
+REGISTER_PROFILER_OPERATION("softmax", "Softmax", profile_softmax);

profiler/src/profiler.cpp

@@ -8,27 +8,7 @@
 
 static void print_helper_message()
 {
-    // clang-format off
-    printf("arg1: tensor operation (gemm: GEMM\n"
-           "                        gemm_splitk: Split-K GEMM\n"
-           "                        gemm_bilinear: GEMM+Bilinear\n"
-           "                        gemm_add_add_fastgelu: GEMM+Add+Add+FastGeLU\n"
-           "                        gemm_reduce: GEMM+Reduce\n"
-           "                        gemm_bias_add_reduce: GEMM+Bias+Add+Reduce\n"
-           "                        batched_gemm: Batched GEMM\n"
-           "                        batched_gemm_gemm: Batched+GEMM+GEMM\n"
-           "                        batched_gemm_add_relu_gemm_add: Batched+GEMM+bias+gelu+GEMM+bias\n"
-           "                        batched_gemm_reduce: Batched GEMM+Reduce\n"
-           "                        grouped_gemm: Grouped GEMM\n"
-           "                        conv_fwd: Convolution Forward\n"
-           "                        conv_fwd_bias_relu: ForwardConvolution+Bias+ReLU\n"
-           "                        conv_fwd_bias_relu_add: ForwardConvolution+Bias+ReLU+Add\n"
-           "                        conv_bwd_data: Convolution Backward Data\n"
-           "                        grouped_conv_fwd: Grouped Convolution Forward\n"
-           "                        grouped_conv_bwd_weight: Grouped Convolution Backward Weight\n"
-           "                        softmax: Softmax\n"
-           "                        reduce: Reduce\n");
-    // clang-format on
+    std::cout << "arg1: tensor operation " << ProfilerOperationRegistry::GetInstance() << std::endl;
 }
 
 int main(int argc, char* argv[])
@@ -37,10 +17,13 @@ int main(int argc, char* argv[])
     {
         print_helper_message();
     }
-    else if(auto operation = ProfilerOperationRegistry::GetInstance().Get(argv[1]); operation.has_value())
+    else if(const auto operation = ProfilerOperationRegistry::GetInstance().Get(argv[1]);
+            operation.has_value())
     {
         return (*operation)(argc, argv);
-    } else {
+    }
+    else
+    {
         std::cerr << "cannot find operation: " << argv[1] << std::endl;
         return EXIT_FAILURE;
     }

profiler/src/profiler_operation_registry.hpp

@@ -2,6 +2,7 @@
 // Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
 
 #include <functional>
+#include <iterator>
 #include <map>
 #include <optional>
 #include <string_view>
@@ -14,7 +15,18 @@ class ProfilerOperationRegistry final
     using Operation = std::function<int(int, char*[])>;
 
     private:
-    std::unordered_map<std::string_view, Operation> operations_;
+    struct Entry final
+    {
+        explicit Entry(std::string_view description, Operation operation) noexcept
+            : description_(description), operation_(operation)
+        {
+        }
+
+        std::string_view description_;
+        Operation operation_;
+    };
+
+    std::map<std::string_view, Entry> entries_;
 
     public:
     static ProfilerOperationRegistry& GetInstance()
@@ -25,22 +37,37 @@ class ProfilerOperationRegistry final
 
     std::optional<Operation> Get(std::string_view name) const
     {
-        const auto found = operations_.find(name);
-        if(found == end(operations_))
+        const auto found = entries_.find(name);
+        if(found == end(entries_))
         {
             return std::nullopt;
         }
 
-        return found->second;
+        return (found->second).operation_;
     }
 
-    bool Add(std::string_view name, Operation operation)
+    bool Add(std::string_view name, std::string_view description, Operation operation)
     {
-        return operations_.try_emplace(name, std::move(operation)).second;
+        return entries_
+            .emplace(std::piecewise_construct,
+                     std::forward_as_tuple(name),
+                     std::forward_as_tuple(description, std::move(operation)))
+            .second;
     }
-};
 
-#define REGISTER_PROFILER_OPERATION(name, operation)                                     \
-    namespace {                                                                          \
-    const bool result = ::ProfilerOperationRegistry::GetInstance().Add(name, operation); \
+    friend std::ostream& operator<<(std::ostream& stream, const ProfilerOperationRegistry& registry)
+    {
+        stream << "{\n";
+        for(auto& [name, entry] : registry.entries_)
+        {
+            stream << "\t" << name << ": " << entry.description_ << "\n";
+        }
+        stream << "}";
+
+        return stream;
     }
+};
+
+#define REGISTER_PROFILER_OPERATION(name, description, operation) \
+    static const bool result =                                    \
+        ::ProfilerOperationRegistry::GetInstance().Add(name, description, operation)