File size: 317 Bytes
b22e03e | 1 2 3 4 5 6 7 8 | # csrc
CUDA extension placeholder for the FlexAttention replacement package.
The v1 package currently exposes the stable Python Tensor API and SDPA-backed
autograd fallback. Native CUDA forward/backward kernels should be added here
without changing the public functions in `torch-ext/flashrt_flex_attention_train`.
|