开发者社区> 问答> 正文

如何优化投机采样中的采样流程以减少额外负担?

如何优化投机采样中的采样流程以减少额外负担?

展开
收起
花开富贵111 2024-05-23 17:40:06 14 0
1 条回答
写回答
取消 提交回答
  • 通过对采样流程进行优化来减少额外负担。具体做法是对存在top_k参数的情况使用融合算子,将原版采样流程分为两步:先进行TopK采样,再省略TopP步骤直接在采样过程中判断。改进后的流程大幅度减少了计算量和kernel数量,从而极大程度减少了采样需要的时间。

    2024-05-23 17:53:06
    赞同 展开评论 打赏
问答地址:
问答排行榜
最热
最新

相关电子书

更多
低代码开发师(初级)实战教程 立即下载
冬季实战营第三期:MySQL数据库进阶实战 立即下载
阿里巴巴DevOps 最佳实践手册 立即下载