diff --git a/graphengine b/graphengine index 433521ddbf..423c0228e8 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 32e8f8844e..a97ecadae0 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") \