Browse Source

!9500 AKG op for MD

From: @mamba_ni
Reviewed-by: @ckey_dou,@gaoxiong1
Signed-off-by: @gaoxiong1
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
2c37dc43b9
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      mindspore/ccsrc/backend/kernel_compiler/kernel_query.cc

+ 4
- 0
mindspore/ccsrc/backend/kernel_compiler/kernel_query.cc View File

@@ -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);


Loading…
Cancel
Save