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.