3 Commits

Author	SHA1	Message	Date
bnellnm	5467ac3196	[Kernel][Misc] Use TORCH_LIBRARY instead of PYBIND11_MODULE for custom ops (#5047)	2024-06-09 16:23:30 -04:00
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
Woosuk Kwon	f0d4e14557	Add fused top-K softmax kernel for MoE (#2769)	2024-02-05 17:38:02 -08:00