4 Commits

Author	SHA1	Message	Date
Tri Dao	aaa1474129	[CrossEntropy] Simplify the case of large vocab with Tensor Parallel	2023-11-19 23:19:36 -08:00
Shijie	abf04a56e1	fix flash ce mp large vocab (#673)	2023-11-19 23:01:07 -08:00
Tri Dao	c79de85ffa	[CrossEntropy] Fix triton cross_entropy_loss IMA for >=2B elements	2023-10-24 00:17:34 -07:00
Tri Dao	5400fdc4ac	[CE] Implement CrossEntropyLoss in Triton	2023-09-15 20:05:28 -07:00