- 博客(8)
- 收藏
- 关注
原创 Megakernel
execute_task():这个函数会根据(task_type, variant_id)找到对应的CUDA kernel调用代码(all_task_variants存着这张图所有要用到的算子的调用和调用参数代码),把task_desc->input_ptrs/output_ptrs传进去跑。EVENT_LAUNCH_MASSIVE_TASKS会按照Scheduler数量,均分任务区间[first_task_id, last_task_id),每个Scheduler拿到的任务区间是均分过的。
2026-09-23 15:52:53
125
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅