Divakar Verma a66cf40b20 [Kernel][ROCm][AMD] enable fused topk_softmax kernel for moe layer (#4927) ...

This PR enables the fused topk_softmax kernel used in moe layer for HIP

2024-06-02 14:13:26 -07:00

1.4 KiB

Raw Blame History

View Raw