liangsu9988's picture
Add package source and torch-universal build variant
b22e03e verified
|
Raw
History Blame Contribute Delete
280 Bytes

torch-ext

PyTorch bindings and Python package for flashrt-flex-attention-train.

torch_binding.cpp registers a package marker so HF kernel-builder can load the extension. The Python module provides the SDPA fallback and the stable training API used by PI052 integration.