MCPcopy Create free account
hub / github.com/NVlabs/LongLive / write_kernel

Function write_kernel

fouroversix/scripts/generate_kernels.py:105–109  ·  view source on GitHub ↗
(kernel: Kernel, autogen_dir: Path)

Source from the content-addressed store, hash-verified

103
104
105def write_kernel(kernel: Kernel, autogen_dir: Path) -> None:
106 prelude = """// Splitting the different transpose modes to different files to speed up compilation.
107// This file is auto-generated. See "generate_kernels.py"\n""" # noqa: E501
108 content = prelude + kernel.template
109 (autogen_dir / kernel.filename).write_text(content)
110
111
112def main(output_dir: str | None) -> None:

Callers 1

mainFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected