Crate pire_base

Crate pire_base 

Source
Expand description

§This crate is only for internal use in the pire project

Nothing is expected to be used outside this module No semver guarantees

Modules§

range_rwlock

Macros§

acc_1x6
acc_1x12
acc_2x6
acc_2x12
acc_3x4
acc_3x8
acc_p_avx
acc_p_avx512
b_num_1x4
b_num_1x6
b_num_1x12
b_num_2x4
b_num_2x6
b_num_2x12
b_num_3x8
b_reg
c_load
c_mem
c_reg_1x4
c_reg_1x6
c_reg_1x12
c_reg_2x4
c_reg_2x6
c_reg_2x12
c_reg_3x4
c_reg_3x8
cum_seq
def_kernel_bb_pf1
def_kernel_bb_v0
def_kernel_bs
def_kernel_sb_pf1
def_kernel_sb_v0
def_pa
def_pire_gemm
def_ukernel_avx512
def_ukernel_neon
def_ukernel_neon_alt
def_ukernel_neon_fp16
def_ukernel_neon_i8mm
def_ukernel_sse
def_ukernel_sve
def_ukernel_sve_i8mm
fmadd_1x12
fmadd_2x12
fmadd_3x8
init_ab
init_ab_2
init_ab_avx
is_mixed
load_a_avx
load_a_avx512
loadp_avx
loadp_avx512
mem
n_cond
packing_api
partial_strided
prefetch_0
prefetch_b
prefetch_c_sse
store_1x6
store_1x12
store_2x6
store_2x12
store_3x4
store_3x8
storep_avx
storep_avx512

Structs§

CpuFeatures
HWConfig
IdentityFn
PackPool
PackedMatrix
PackedMatrixMixed
PirePar
PireThreadConfig
PoolSize
StridedMatrix
StridedMatrixMut
StridedMatrixP

Enums§

Array
ArrayMut
GemmPool
HWModel
PArray
PArrayMixed
PtrData

Constants§

AB_ALIGN

Statics§

PACK_POOL
PIRE_NUM_THREADS
RUNTIME_HW_CONFIG

Traits§

BaseNum
GemmCache
UnaryFn

Functions§

acquire
avx_vzeroupper
extend
get_apbp_barrier
get_cache_params
get_mem_pool_size_goto
get_mem_pool_size_small_m
get_mem_pool_size_small_n
has_c32_compute
has_c64_compute
has_f16_compute
has_f16f32_compute
has_f32_compute
has_f64_compute
has_i8i32_compute
has_i16i32_compute
hw_model
load_buf
matrix_size
matrix_size_strided
pire_num_threads
run_small_m
run_small_n
split_c_range
split_range
store_buf