[LLGA] add profileName for LlgaKernel (#40)

Author: chunyuan-w

torch_ipex/csrc/jit/codegen/onednn/interface.cpp

@@ -49,7 +49,7 @@ void fuseGraph(std::shared_ptr<Graph>& g) {
 Operation createLlgaKernel(const Node* node) {
   auto kernel = std::make_shared<fuser::onednn::LlgaKernel>(node);
   return [kernel](Stack* stack) {
-    RECORD_FUNCTION(kernel->debugName(), std::vector<c10::IValue>());
+    RECORD_FUNCTION(kernel->profileName(), std::vector<c10::IValue>());
     kernel->run(*stack);
     return 0;
   };

torch_ipex/csrc/jit/codegen/onednn/kernel.cpp

@@ -55,7 +55,8 @@ LlgaKernel::LlgaKernel(const Node* fusionNode)
       graph_(fusionNode->g(attr::Subgraph)),
       nInputs_(graph_->inputs().size()),
       nOutputs_(graph_->outputs().size()),
-      debugName_(genDebugName()) {
+      debugName_(genDebugName()),
+      profileName_(genProfileName()) {
   // TODO: This is a workaround to recreate the partitions here. 
   // The ideal way is to use the partition serialization API (not available from LLGA now) 
   // to carry a serialized string representation from graph rewrite and deserialize it here.

torch_ipex/csrc/jit/codegen/onednn/kernel.h

@@ -29,6 +29,10 @@ class LlgaKernel {
     return debugName_;
   }
 
+  const std::string& profileName() const {
+    return profileName_;
+  }
+
  private:
   bool useOpaqueLayout(size_t offset) const;
 
@@ -57,6 +61,16 @@ class LlgaKernel {
     return "LlgaPartition_" + std::to_string(debugId++);
   }
 
+  std::string genProfileName() {
+    std::vector<std::string> op_list;
+    for (auto* node : graph_->block()->nodes()) {
+      if (node->kind().is_aten()) {
+        op_list.push_back(node->kind().toUnqualString());
+      }
+    }
+    return c10::Join("+", op_list);
+  }
+
   static dnnl::graph::logical_tensor toLogicalTensor(const ArgSpec& s) {
     return s.logical_tensor();
   }
@@ -68,6 +82,7 @@ class LlgaKernel {
   int64_t nOutputs_ = 0;
   dnnl::graph::partition partition_;
   std::string debugName_;
+  std::string profileName_;
 };
 
 } // namespace onednn