online-softmax:“ softmax的在线规范化器计算”文章的基准代码

基准 这是通过在线计算softmax归一化项来探索Softmax和Softmax + TopK函数性能的基准。 请参阅 。 建造 您将需要 v1.8.0（或更高版本）和CUDA 9.1（或更高版本）来建立基准。 将Makefile中的CUB变量设置为CUB库的正确位置。 建立样本：make 跑步 运行示例：./online_softmax_benchmark