AttributeError: module 'tvm.contrib.graph_runtime' has no attribute 'GraphModule

Hi, I have installed tvm from source and have followed all the installation steps. I am using llvm - 6.0 and python 3.7 on a 32 bit ubuntu system. When i try to create a module such as below:

gmod = graph_runtime.GraphModule(lib["default"](ctx))

i get this error:

AttributeError: module 'tvm.contrib.graph_runtime' has no attribute 'GraphModule'

i tried building it by source again, but without any success. Does anyone know what went wrong and how i can solve it?

1 Like

There was an API change. Try the following:

Change:

from tvm.contrib import graph_runtime

to:

import tvm.contrib.graph_executor as runtime

Also change graph_runtime to runtime.

Hope it works for you.

2 Likes

Thanks a lot @alopez_13 ! It worked