src/pygpukit/ops/
└── matmul.py (1981 lines - everything)
src/pygpukit/ops/matmul/
├── __init__.py (exports)
├── generic.py (matmul, batched_matmul, transpose)
├── fp8.py (FP8 GEMM variants)
├── nvf4.py (NVF4 GEMM)
├── gemv.py (GEMV operations)
├── grouped.py (grouped GEMM for MoE)
├── quantize.py (quantization helpers)
└── availability.py (all *_available functions)
Problem
src/pygpukit/ops/matmul.pyis 1981 lines with mixed concerns.Current State
Proposed Structure
Benefits
Related