C++ equivalent of Python IRBuilder

hlu1 · March 27, 2019, 10:54pm

Is there a C++ equivalent of Python IRBuilder (https://github.com/dmlc/tvm/blob/master/python/tvm/ir_builder.py)? I would like to port this function to C++ which uses IRBuilder with assembly microkernels:

github.com

ajtulloch/tvm/blob/sparse-ops/topi/python/topi/x86/sparse.py#L236-L248


def sparse_length_sum_fused_8_bit_rowwise_ir(data, indices, lengths, out):
    irb = tvm.ir_builder.create()
    data_ptr = irb.buffer_ptr(data)
    indices_ptr = irb.buffer_ptr(indices)
    lengths_ptr = irb.buffer_ptr(lengths)
    assert not data.strides
    assert not out.strides
    out_ptr = irb.buffer_ptr(out)
    irb.scope_attr(tvm.const(1), "pragma_import_llvm", sls_fused_impl())
    irb.emit(tvm.call_extern(
            "int32", "embedding_lookup",
            B - 8, L, I, data_ptr, indices_ptr, lengths_ptr, out_ptr))
    return irb.get()