fastvideo-kernel / README.md
sayakpaul's picture
sayakpaul HF Staff
Uploaded using `kernel-builder`.
8ca1c7e verified
|
Raw
History Blame Contribute Delete
1.03 kB
metadata
library_name: kernels
license: apache-2.0

This is the repository card of sayakpaul/fastvideo-kernel that has been pushed on the Hub. It was built to be used with the kernels library. This card was automatically generated.

How to use

# make sure `kernels` is installed: `pip install -U kernels`
from kernels import get_kernel

kernel_module = get_kernel("sayakpaul/fastvideo-kernel", version=1)
sliding_tile_attention = kernel_module.sliding_tile_attention

sliding_tile_attention(...)

Available functions

  • sliding_tile_attention
  • video_sparse_attn
  • video_sparse_attn_bshd
  • block_sparse_attn
  • block_sparse_attn_from_indices
  • moba_attn_varlen
  • process_moba_input
  • process_moba_output
  • Int8Linear
  • FastRMSNorm
  • FastLayerNorm
  • int8_linear
  • int8_quant
  • __version__

Benchmarks

No benchmark available yet.

Upstream

The original source code for this kernel comes from https://github.com/hao-ai-lab/FastVideo.