From f062532d592273e8ee4b6330f356da613de84374 Mon Sep 17 00:00:00 2001
From: jjfeing <jiangjianfei3@huawei.com>
Date: Tue, 7 Mar 2023 17:05:33 +0800
Subject: [PATCH] revert code

---
 mindspore/ccsrc/plugin/device/ascend/kernel/rts/label_switch.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/plugin/device/ascend/kernel/rts/label_switch.cc b/mindspore/ccsrc/plugin/device/ascend/kernel/rts/label_switch.cc
index ef444ee8286..540928cfc6e 100644
--- a/mindspore/ccsrc/plugin/device/ascend/kernel/rts/label_switch.cc
+++ b/mindspore/ccsrc/plugin/device/ascend/kernel/rts/label_switch.cc
@@ -73,7 +73,7 @@ std::vector<TaskInfoPtr> LabelSwitchKernel::GenTask(const std::vector<AddressPtr
 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> LabelSwitchDesc::GetKernelInfo(const CNodePtr &) {
   std::vector<std::shared_ptr<kernel::KernelBuildInfo>> label_switch_build_info{};
   std::vector<string> input_format{kOpFormat_DEFAULT};
-  std::vector<TypeId> input_type{kNumberTypeUInt64};
+  std::vector<TypeId> input_type{kNumberTypeInt32};
   if (input_format.size() != input_type.size()) {
     MS_LOG(EXCEPTION) << "Invalid param num, input_format size " << input_format.size() << " input_type size "
                       << input_type.size();