cubecl-llvm 0.11.0-pre.4

LLVM compiler for CubeCL
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

pub use crate::{
    shared::{
        intrinsic::{call_op, i32_const_op, i32_ty},
        lowering::TargetLowering,
        metadata::EntryArgLayout,
        polyfill::LowerOp,
        shared_memory::SharedDeclarations,
        to_llvm::{
            CubeToLLVMType, ToLLVMDialect,
            constant::{
                I32_WIDTH, constant_op, convert_attr, float_attr, insert_bool_const,
                insert_i32_const, insert_int_const, int_attr,
            },
            insert::insert,
            ty::{
                GEP_INDEX_WIDTH, cube_type_to_llvm, index_width, llvm_mangled_ty, scalar_alignment,
                type_alignment,
            },
            vector::insert_splat,
        },
    },
    target::{CtxTarget, LlvmTarget},
};
pub use cubecl_core as cubecl;
pub use cubecl_core::ir::{
    AddressSpace, Builtin, NamedRewrite, OpInserter, Scope,
    attributes::{BufferIOAttr, EntrypointInterface, FuncInterface, IndexAttr, ZeroAttr},
    interfaces::{
        AlignedType, MaterializableOp, ScalarType, ScalarizableType, SizedType, TypedExt,
    },
    prelude::*,
    settings::Dim3,
    types::{
        ArrayType as CubeArrayType, PointerType as CubePointerType, VectorType as CubeVectorType,
        scalar::{BoolType, IndexType},
    },
};
pub use pliron::{
    attribute::AttrObj,
    basic_block::BasicBlock,
    builtin::{
        attributes::{FPDoubleAttr, FPHalfAttr, FPSingleAttr, IntegerAttr},
        ops::{ConstantOp, FuncOp, ModuleOp},
        types::{FP16Type, FP32Type, FP64Type, FunctionType, IntegerType, Signedness},
    },
    identifier::Identifier,
    input_err,
    irbuild::inserter::{BlockInsertionPoint, OpInsertionPoint},
    printable::Printable,
    symbol_table::SymbolTableCollection,
    utils::apint::{APInt, bw},
};
pub use pliron_llvm::{
    attributes::{
        AtomicOrderingAttr, AtomicRmwKindAttr, FCmpPredicateAttr, FastmathFlags, FastmathFlagsAttr,
        ICmpPredicateAttr, IntegerOverflowFlagsAttr, LinkageAttr, SyncScopeAttr,
    },
    op_interfaces::{
        AlignableOpInterface, BinArithOp, CastOpInterface, CastOpWithNNegInterface, FastMathFlags,
        FloatBinArithOpWithFastMathFlags, IntBinArithOpWithOverflowFlag,
    },
    ops as llvm,
    types::{
        ArrayType as LlvmArrayType, FuncType, PointerType as LlvmPointerType,
        VectorType as LlvmVectorType, VectorTypeKind, VoidType,
    },
};
pub use thiserror::Error;