[tvm] branch main updated: [Releay] Fix on_device call for explicit virtual_device (#12088)

[masahi]

This is an automated email from the ASF dual-hosted git repository.

masahi pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 2fee86e43b [Releay] Fix on_device call for explicit virtual_device 
(#12088)
2fee86e43b is described below

commit 2fee86e43ba660af3a5fbd0213a8ea4178fd53b2
Author: Andrey Malyshev <elvin.n...@gmail.com>
AuthorDate: Thu Jul 14 22:21:27 2022 +0300

    [Releay] Fix on_device call for explicit virtual_device (#12088)
---
 python/tvm/relay/op/annotation/annotation.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/tvm/relay/op/annotation/annotation.py 
b/python/tvm/relay/op/annotation/annotation.py
index 5582ac9981..685a8807f7 100644
--- a/python/tvm/relay/op/annotation/annotation.py
+++ b/python/tvm/relay/op/annotation/annotation.py
@@ -28,6 +28,8 @@ def _make_virtual_device(device):
         return target.VirtualDevice(device)
     if isinstance(device, str):
         return target.VirtualDevice(_nd.device(device))
+    if isinstance(device, target.VirtualDevice):
+        return device
     raise ValueError("expecting a Device or device name, but received a %s" % 
(type(device)))