Crash triggered by 5 lines of TIR code

UniverseFly · August 16, 2021, 10:20am

Hi community, I have been playing with tvm for several months and been engaged in developing some low level operators. However, when I try to implement some feature related to the Broadcast node, an unexpected crash happens:

from tvm import tir
import tvm

if __name__ == '__main__':
    a = tir.Broadcast(tir.const(1), 2)
    v = tir.Var('i1', 'int32')
    stmt = tir.Store(v, a, a, None)
    func = tir.PrimFunc([v], stmt)
    tvm.build(func)

This gives me a segmentation fault. And gdb tells it may be related to visiting StoreNode:

Thread 1 "python" received signal SIGSEGV, Segmentation fault.
0x00007fffde544818 in tvm::codegen::CodeGenLLVM::VisitStmt_(tvm::tir::StoreNode const*) ()

I lack experience in debugging TVM and I would be grateful if anyone can help me