extensions/softmax/scaled_upper_triangle_masked_softmax_cuda.py · 7cfed5f076f683293eb559c7e61feb6d3376b8d3 · OpenDAS / ColossalAI

Frank Lee authored Jan 25, 2024

* [feat] refactored extension module

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

7cfed5f0

scaled_upper_triangle_masked_softmax_cuda.py 1.07 KB