MCPcopy Create free account
hub / github.com/RightNow-AI/autokernel / _get_module

Function _get_module

kernels/cuda/flash_attention.py:241–245  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

239
240
241def _get_module():
242 global _module
243 if _module is None:
244 _module = compile_cuda(CUDA_SRC, "flash_attention_cuda")
245 return _module
246
247
248def kernel_fn(

Callers 1

kernel_fnFunction · 0.70

Calls 1

compile_cudaFunction · 0.90

Tested by

no test coverage detected