From 740d5bac670b2d4f50b11f60a99bcbc0c245c2bb Mon Sep 17 00:00:00 2001 From: lilei Date: Thu, 29 Oct 2020 19:57:20 +0800 Subject: [PATCH] modify_version info for 1.1.0 --- graphengine | 2 +- mindspore/ops/_op_impl/tbe/top_k.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/graphengine b/graphengine index 433521ddbf3..423c0228e8c 160000 --- a/graphengine +++ b/graphengine @@ -1 +1 @@ -Subproject commit 433521ddbf33918a86d0a16bafa200c95baa4eea +Subproject commit 423c0228e8c421f2b095e40d14e9fb3b563f63aa diff --git a/mindspore/ops/_op_impl/tbe/top_k.py b/mindspore/ops/_op_impl/tbe/top_k.py index 32e8f8844e3..a97ecadae06 100644 --- a/mindspore/ops/_op_impl/tbe/top_k.py +++ b/mindspore/ops/_op_impl/tbe/top_k.py @@ -19,9 +19,9 @@ from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType top_k_op_info = TBERegOp("TopK") \ .fusion_type("OPAQUE") \ .async_flag(False) \ - .binfile_name("top_k.so") \ + .binfile_name("top_k_d.so") \ .compute_cost(10) \ - .kernel_name("top_k") \ + .kernel_name("top_k_d") \ .partial_flag(True) \ .attr("dim", "optional", "int", "all") \ .attr("k", "required", "int", "all") \