ViperEkura 2bc4d2b8a8 refactor: unify kernel module loading and packaging ...

- loader.py: lazy/cached import; is_available defers the actual load; get_module raises on unavailable
- ops/{attention,rotary,fp8}: use get_module instead of touching private _modules or their own _mod() cache
- package-data: ship astrai.extension.lib *.so in built wheels (non-editable installs previously lost every kernel)

2026-08-23 15:57:04 +08:00

..

backend

refactor: rework attention backend resolution

2026-08-23 14:47:02 +08:00

lib

refactor: move compiled kernel .so files into extension/lib

2026-07-31 15:36:32 +08:00

ops

refactor: unify kernel module loading and packaging

2026-08-23 15:57:04 +08:00

__init__.py

refactor: separate extension ops and backends

2026-08-16 21:15:52 +08:00

fp8.py

perf: pure FP8 fwd/bwd and lean non-transposed GEMM

2026-08-23 15:38:30 +08:00

loader.py

refactor: unify kernel module loading and packaging

2026-08-23 15:57:04 +08:00