builtin.py

"""The language interface for tl programs."""

from tvm import tir


def CreateListofMBarrierOp(*args):
    return tir.call_intrin("handle", tir.op.Op.get("tl.CreateListofMBarrierOp"), *args)


def GetMBarrierOp(*args):
    return tir.call_intrin("handle", tir.op.Op.get("tl.GetMBarrierOp"), *args)


def CreateTMADescriptorOp(*args):
    return tir.call_intrin("handle", tir.op.Op.get("tl.CreateTMADescriptorOp"), *args)


def TMALoadOp(*args):
    return tir.call_intrin("handle", tir.op.Op.get("tl.TMALoadOp"), *args)


def FenceProxyAsyncOp(*args):
    return tir.call_intrin("handle", tir.op.Op.get("tl.FenceProxyAsyncOp"), *args)


def TMAStoreArrive(*args):
    return tir.call_intrin("handle", tir.op.Op.get("tl.TMAStoreArrive"), *args)


def TMAStoreWait(*args):
    return tir.call_intrin("handle", tir.op.Op.get("tl.TMAStoreWait"), *args)


def SetMaxNReg(*args):
    return tir.call_intrin("handle", tir.op.Op.get("tl.SetMaxNReg"), *args)


def NoSetMaxNReg(*args):
    return tir.call_intrin("handle", tir.op.Op.get("tl.NoSetMaxNReg"), *args)


def MBarrierWaitParity(*args):
    return tir.call_intrin("handle", tir.op.Op.get("tl.MBarrierWaitParity"), *args)


def MBarrierExpectTX(*args):
    return tir.call_intrin("handle", tir.op.Op.get("tl.MBarrierExpectTX"), *args)


def WaitWgmma(*args):
    return tir.call_intrin("handle", tir.op.Op.get("tl.WaitWgmma"), *args)