ViperEkura
16a55bb474
- move attention kernels to csrc/kernels/attention/ and rotary to rotary/ - add shared common/mma.cuh (mma_sync, ldmatrix) and device.cuh (sm checks) - split fp8_mm into three-layer fp8/common.h, gemm.cuh, mm.cu - fix fused FP8 GEMM ldmatrix lane indexing to fix OOB shared reads - update extension ops, loader, and kernel tests
3 lines
135 B
Python
3 lines
135 B
Python
# Source directory for CUDA kernels — build-time only.
|
|
# Compiled .so files live in astrai/extension/lib/ (see csrc/CMakeLists.txt).
|