From 3d4963b818812099207e0060c8cb98a02f6142f1 Mon Sep 17 00:00:00 2001 From: Xuxue1 Date: Wed, 13 May 2020 12:27:14 +0800 Subject: [PATCH] Add prim::device op --- 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 d95a9122ad59..967b10894f15 100644 --- a/python/tvm/relay/frontend/pytorch.py +++ b/python/tvm/relay/frontend/pytorch.py @@ -1609,6 +1609,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"),