operator_task_package/flashattn_task_package

FlashAttention Task Package

本目录用于 FlashAttention KV Cache Decode 题目的题面归档、基准理解、提交示例和算子优化。

目录职责

flashattn_task_package/
├── benchmark/                         FlashAttention API 基准脚本
├── starter/                           OJ 提供的各语言冒烟示例原件
├── xpuoj_problem/                     当前正式题面
└── kernel_ops/
    └── FlashAttentionKVCacheDecode/   实际优化工作区
        ├── tilelang/                  TileLang 编号版本与当前提交代码
        ├── cudamaca/                  CUDA MACA 编号版本
        ├── triton/                    Triton 编号版本
        ├── tests/                     正确性、性能和 profiling 驱动
        ├── results/                   原始日志、JSON、CSV 和 OJ 结果
        └── workflows/
            ├── prompts/               可复用 Agent Prompt
            └── references/            本题教程和专用参考资料

信息优先级

1. xpuoj_problem/Agent 推理算子库优化 - FlashAttention KV Cache Decode.md 是接口、数据范围和正确性要求的唯一正式依据。
2. starter/ 中的文件只用于理解提交形式，不保证覆盖当前题面的全部语义，也不代表性能基线。
3. benchmark/ 用于理解 FlashAttention API 和性能趋势，其参数与当前 OJ 测试范围不完全一致。
4. kernel_ops/FlashAttentionKVCacheDecode/ 保存实际候选代码、测试、结果和优化记录。

公共知识库

跨算子复用的架构、TileLang、性能分析和实验方法统一维护在 /data/operator_task_package/ref。本题目录只保留本题专属材料和原始实验产物，避免复制后产生不一致版本。

建议入口：

• /data/operator_task_package/ref/08_templates/new_operator_checklist.md
• /data/operator_task_package/ref/01_methodology/operator_optimization_lifecycle.md
• /data/operator_task_package/ref/03_backends/tilelang/programming_and_compilation.md
• /data/operator_task_package/ref/04_profiling/profiling_workflow.md
• /data/operator_task_package/ref/05_optimization_patterns/