MCPcopy
hub / github.com/huggingface/transformers / load_cuda_kernels

Function load_cuda_kernels

src/transformers/models/yoso/modeling_yoso.py:51–58  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

49
50
51def load_cuda_kernels():
52 global lsh_cumulation
53 if not is_kernels_available():
54 raise ImportError("kernels is not installed, please install it with `pip install kernels`")
55 from ...integrations.hub_kernels import get_kernel
56
57 yoso = get_kernel("kernels-community/yoso")
58 lsh_cumulation = yoso.lsh_cumulation
59
60
61def to_contiguous(input_tensors):

Callers 1

__init__Method · 0.70

Calls 2

is_kernels_availableFunction · 0.85
get_kernelFunction · 0.85

Tested by

no test coverage detected