wenyi-li 中等规模 1D Amin 归约优化
任务特征
- 数据尺寸：(65536,)，中等规模1D数据
优化：适中并行度
# （AI core=40）
# 1. grid=4<40 -> 2.47 us
triton.Config({'BLOCK_SIZE': 16384})
# 2. grid=8<40 -> 2.21 us 最优
triton.Config({'BLOCK_SIZE': 8192})
# 3. grid=32<40 -> 2.92 us
triton.Config({'BLOCK_SIZE': 2048})
# 4. grid=40 -> 3.44 us
triton.Config({'BLOCK_SIZE': 1639})
# 5. grid=128>40 -> 6.70 us
triton.Config({'BLOCK_SIZE': 512})
总结
中等规模数据，网格数适当小时性能最佳。存在最优并行度平衡点：网格数过小导致单块负载过重，过大则引入过多调度开销。