File size: 590 Bytes
ccef021 | 1 2 3 4 5 6 7 8 9 10 11 12 | 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
|