
【CS336】lecture6 占用率|bank冲突|Triton|GeLU算子|softmax算子|reduce算子|matmul算子
回顾这节主要讲的是如何用triton写算子,优化性能。首先回顾一下上一节讲的GPU架构。一个GPU上有多个SM,多个SM共享全局内存...

回顾这节主要讲的是如何用triton写算子,优化性能。首先回顾一下上一节讲的GPU架构。一个GPU上有多个SM,多个SM共享全局内存...
摘要:在人脸识别、行人重识别、图像检索等度量学习任务中,传统Softmax损失函数早已无法满足高精度特征匹配需求。Softmax仅能...
Softmax在YOLOv8中只做一件事——把16个Bin的分数变成概率 一句话总结:Softmax 在 YOLOv8 中只用于 DFL 解码——...