Download flashmla_tests/kernelkit/__init__.py from jiliu1/avo_test_cases: direct link, hf CLI and curl.
- Browser
- Download file 590 Bytes
-
https://huggingface.co/jiliu1/avo_test_cases/resolve/main/flashmla_tests/kernelkit/__init__.py
- Command line
-
hf download hf://jiliu1/avo_test_cases/flashmla_tests/kernelkit/__init__.py
-
curl -L -o __init__.py https://huggingface.co/jiliu1/avo_test_cases/resolve/main/flashmla_tests/kernelkit/__init__.py
590 Bytes
| 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 | |