TVM on Jetson TX2 (GPU)

Hi,

I am trying to run resnet18 on TX2 (GPU). On CPU it is working fine but on GPU, I am getting a seg fault.

Below is the host and target assignment. I suspect it may not be correct.

target = tvm.target.Target("nvidia/jetson-tx2", host="llvm -mtriple=aarch64-linux-gnu")
dev = tvm.cuda(0)

with tvm.transform.PassContext(opt_level=3):
    lib = relay.build(mod, target=target, params=params)

m = graph_executor.GraphModule(lib["default"](dev))

The last line is causing the Segmentation fault Is there a different way, I should be providing the target and host?

I have tried target = tvm.target.Target("cuda", host="llvm") and target = tvm.target.Target("cuda -tx2", host="llvm")

Thank you.