[BYOC]symbol lookup:error /tmp/...../lib.so undefined symbol

heart1998 · July 28, 2022, 6:59am

Blockquote 2022-07-27T16:00:00Z

Hello, I use BYOC to add XPU to my TVM, link the handwritten network with the dynamic link library generated by tvm, and run the network training, this problem occurs, my configuration is TVM0.8dev.

/home/qwe/anaconda3/envs/torch/bin/python: symbol lookup error: /tmp/tvm-debug-mode-tempdirs/2022-07-24T13-18-44___jp3n0owg/00000/lib.so: undefined symbol: _ZN5baidu3xpu3api3tvm7shapeofEiiPiRKSt6vectorIiSaIiEE

chenugray · July 28, 2022, 12:30pm

maybe you could read the code about how invocation external api

tvm support using external library like cudnn.

github.com

apache/tvm/blob/ee319d9d23c80091da9c4fb764b1e6d49d462714/python/tvm/contrib/cudnn.py#L665


            conv_dtype,
            groups,
        ),
        name="y",
    )


return te.extern(
    oshape,
    [x, w],
    lambda ins, outs: tvm.tir.call_packed(
        "tvm.contrib.cudnn.conv3d.forward",
        conv_mode,
        tensor_format,
        algo,
        pad[0],
        pad[1],
        pad[2],
        stride[0],
        stride[1],
        stride[2],
        dilation[0],

I think it’s helpful for you if you are using baidu xpu.