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

Function _get_module

kernels/cuda/rotary_embedding.py:150–154  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

148
149
150def _get_module():
151 global _module
152 if _module is None:
153 _module = compile_cuda(CUDA_SRC, "rotary_embedding_cuda")
154 return _module
155
156
157def _get_freqs(N: int, D: int, device: torch.device):

Callers 1

kernel_fnFunction · 0.70

Calls 1

compile_cudaFunction · 0.90

Tested by

no test coverage detected