File size: 436 Bytes
8e9f35a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | """SM120 kernel recipe."""
from .mainloop import SolAttnForwardSm120
def make_kernel(
*,
debug_route_trace: bool = False,
prefetch_first_exact_k: bool = True,
prefetch_next_route_k: bool = True,
):
return SolAttnForwardSm120(
debug_route_trace=debug_route_trace,
prefetch_first_exact_k=prefetch_first_exact_k,
prefetch_next_route_k=prefetch_next_route_k,
)
__all__ = ["make_kernel"]
|