tvm-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From Anand Venkat via TVM Discuss <>
Subject [TVM Discuss] [Development] Upstreaming tensorize implementation
Date Fri, 26 Apr 2019 08:03:13 GMT

Wanted to add another problem I'm facing. When I call tensorize with call_extern within a
AutoTVM template function it is not able to resolve the external call. But the problem does
not appear when the function is not decorated as a AutoTVM template function. I load the dynamic
library via 

def load_lib():
    """Load library, the functions will be registered into TVM"""
    curr_path = os.path.dirname(os.path.abspath(os.path.expanduser(__file__)))
    # load in as global so the global extern symbol is visible to other dll.
    lib = ctypes.CDLL(
        os.path.join(curr_path, "../"), ctypes.RTLD_GLOBAL)
    return lib

_LIB = load_lib()

in the script that I invoke. contains my external function called with call_extern
but is not resolved. Says symbol lookup error. The script uses tensorize , and one of the
intrinsic function is implmented by call_extern to a function within the library.

[Visit Topic]( to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click here](

Tianqi Chen, UW, Seattle, WA, 98105, United States
  • Unnamed multipart/alternative (inline, None, 0 bytes)
View raw message