ninetoothed.build

ninetoothed.build(premake, configs, *, caller=None, kernel_name=None, output_dir=None)

Build a kernel from a premake function and configs.

Parameters:

• premake – A callable that returns the arrangement, application, and tensors for a given configuration.

• configs – An iterable of configurations where each configuration is a tuple of (args, kwargs, compilation_configs). args and kwargs are passed to premake, and compilation_configs contains compilation configurations for ninetoothed.make (e.g., num_warps and num_stages).

• caller – Who will call the compute kernel.

• kernel_name – The name for the generated kernel.

• output_dir – The directory to store the generated files.