test_llvm.py

from xdsl.dialects import llvm, builtin, arith


def test_llvm_pointer_ops():
    module = builtin.ModuleOp.from_region_or_ops([
        idx := arith.Constant.from_int_and_width(0, 64),
        ptr := llvm.IntToPtrOp.get(idx, ptr_type=builtin.i32),
        llvm.LoadOp.get(ptr),
        nullptr := llvm.NullOp.get(),
        alloc_ptr := llvm.AllocaOp.get(idx, elem_type=builtin.IndexType()),
        llvm.LoadOp.get(alloc_ptr),
    ])

    module.verify()

    assert len(alloc_ptr.res.uses) == 1
    assert ptr.input is idx.result
    assert isinstance(ptr.output.typ, llvm.LLVMPointerType)
    assert ptr.output.typ.type == builtin.i32
    assert isinstance(ptr.output.typ.addr_space, builtin.NoneAttr)

    assert isinstance(nullptr.nullptr.typ, llvm.LLVMPointerType)
    assert isinstance(nullptr.nullptr.typ.type, builtin.NoneAttr)
    assert isinstance(nullptr.nullptr.typ.addr_space, builtin.NoneAttr)


def test_llvm_pointer_type():

    assert llvm.LLVMPointerType.typed(builtin.i64).type is builtin.i64
    assert isinstance(
        llvm.LLVMPointerType.typed(builtin.i64).addr_space, builtin.NoneAttr)

    assert isinstance(llvm.LLVMPointerType.untyped().type, builtin.NoneAttr)
    assert isinstance(llvm.LLVMPointerType.untyped().addr_space,
                      builtin.NoneAttr)