akg ops for md

format fix fix bug
4 years ago · 7c6fc3accc
parent 77587bb04d
commit 7c6fc3accc
1 changed files with 4 additions and 0 deletions
--- a/mindspore/ccsrc/backend/kernel_compiler/kernel_query.cc
+++ b/mindspore/ccsrc/backend/kernel_compiler/kernel_query.cc
@ -107,6 +107,10 @@ void KernelQuery(const CNodePtr &kernel_node, std::vector<std::shared_ptr<kernel
  auto context_ptr = MsContext::GetInstance();
  MS_EXCEPTION_IF_NULL(context_ptr);

+  const PrimitivePtr kPrimProdForceSeA = std::make_shared<Primitive>("ProdForceSeA");
+  if (IsPrimitiveCNode(kernel_node, kPrimProdForceSeA)) {
+    kernel_type = KernelType::AKG_KERNEL;
+  }
  switch (kernel_type) {
    case KernelType::AKG_KERNEL:
      AkgMetadataInfo(kernel_node, kernel_info_list);