forked from ccf-ai-infra/TileOPs-Metax
FrRay
c3a5d60998
• 新增 GroupCountFwdOp 与 TileLang GroupCountFwdKernel • 使用共享内存局部计数和全局原子归并统计路由分组 • 支持 int64 路由索引、-1 padding 和动态 token 数量 • 将 Manifest 状态更新为 implemented 并登记 kernel map • 补充正确性、边界和异常输入测试 • 新增基于 ManifestBenchmark 的 torch.bincount 性能基准 |
||
|---|---|---|
| .. | ||
| kernels | ||
| ops | ||
| perf | ||
| trace | ||
| __init__.py | ||
| compile_contract.py | ||
| conftest.py | ||
| nsa_utils.py | ||
| test_base.py | ||
| test_compile.py | ||
| test_op_base.py | ||
| test_ops_manifest.py | ||
| test_reclaim_action.py | ||
| test_roofline_codegen.py | ||
| test_tier_validation.py | ||
| test_validate_manifest.py | ||