From 293d7197365b4d96e6be62e107b49244501769df Mon Sep 17 00:00:00 2001
From: Candy <1915998056@qq.com>
Date: Wed, 13 May 2020 15:07:31 +0800
Subject: [PATCH] Add prim::device op (#5584)

---
 python/tvm/relay/frontend/pytorch.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/tvm/relay/frontend/pytorch.py b/python/tvm/relay/frontend/pytorch.py
index c7eccf760b5e..080046bfd990 100644
--- a/python/tvm/relay/frontend/pytorch.py
+++ b/python/tvm/relay/frontend/pytorch.py
@@ -1615,6 +1615,7 @@ def _wrap_const(c):
 def _get_convert_map(prelude):
     convert_map = {
         "aten::device"                          : _none(),
+        "prim::device"                          : _none(),
         "aten::sub"                             : _elemwise("subtract"),
         "aten::sub_"                            : _elemwise("subtract"),
         "aten::max"                             : _elemwise("maximum"),