From aa8051a9e89d6af8744cf856a79e4018fdb8767f Mon Sep 17 00:00:00 2001
From: mengyuanli <limengyuan4@huawei.com>
Date: Fri, 7 May 2021 10:53:50 +0800
Subject: [PATCH] check device list size in context

---
 mindspore/lite/src/inner_context.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/mindspore/lite/src/inner_context.cc b/mindspore/lite/src/inner_context.cc
index a3178bc8cc..4e2c036dde 100644
--- a/mindspore/lite/src/inner_context.cc
+++ b/mindspore/lite/src/inner_context.cc
@@ -104,6 +104,10 @@ int InnerContext::IsValid() const {
     MS_LOG(ERROR) << "Device list is empty.";
     return RET_NOT_SUPPORT;
   }
+  if (this->device_list_.size() > 2) {
+    MS_LOG(ERROR) << "Not support device list more than 2.";
+    return RET_NOT_SUPPORT;
+  }
   if (!IsUserSetCpu()) {
     MS_LOG(ERROR) << "CPU context should be set.";
     return RET_NOT_SUPPORT;