flash-mla / tests /kernelkit /__init__.py
medmekk's picture
Upload folder using huggingface_hub
ccef021 verified
from . import bench
from . import compare
from . import generate
from . import precision
from . import utils
from .bench import bench_kineto, bench_by_cuda_events
from .compare import get_cos_diff, check_is_bitwise_equal, check_is_allclose, check_is_bitwise_equal_comparator, check_is_allclose_comparator
from .generate import gen_non_contiguous_randn_tensor, gen_non_contiguous_tensor, non_contiguousify
from .precision import LowPrecisionMode, is_low_precision_mode, optional_cast_to_bf16_and_cast_back
from .utils import colors, cdiv, is_using_profiling_tools, set_random_seed, Counter