From 813dd2f1c0845e59523f78d5d1cc6c281696bb05 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Tue, 11 May 2021 14:23:28 +0800
Subject: [PATCH] fix(mge/core): support non-Tensor args for
 `core.tensor.utils.get_device`

GitOrigin-RevId: 51dc03db9ed4bacc3f4bd0a5f9fd2be914605ed6
---
 imperative/python/src/tensor.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/imperative/python/src/tensor.cpp b/imperative/python/src/tensor.cpp
index 1b16fe56..5a3da411 100644
--- a/imperative/python/src/tensor.cpp
+++ b/imperative/python/src/tensor.cpp
@@ -764,7 +764,7 @@ CompNode _get_device(PyObject*const* args, size_t nargs) {
         }
     }
     if (!valid) {
-        mgb_assert(0, "expect at least 1 device");
+        return CompNode::load(get_default_device());
     }
     Py_XDECREF(tuple);
     return cn;
-- 
GitLab