3 Commits

Author	SHA1	Message	Date
Awni Hannun	4f72c66911	improvements to scatter / gather (#1541)	2024-10-30 19:30:54 -07:00
Vijay Krish	972d9a3aea	Up to 10x faster scatter. (#709) ... * Faster scatter. Add specialization for 1-d index tensors. * Address review comments. - Check for row contiguity of index, update tensors instead of checking strides. - Add support for 1d specialization with col contiguous update tensor, along with a test. * Nit1 Co-authored-by: Awni Hannun <awni.hannun@gmail.com> * Nit2 Co-authored-by: Awni Hannun <awni.hannun@gmail.com> --------- Co-authored-by: Awni Hannun <awni.hannun@gmail.com>	2024-02-21 11:09:30 -08:00
Vijay Krish	06072601ce	Scatter optimization : Eliminate 64b integer divide. (#662) ... Launch 2D grid to eliminate divide and mod in device code, since 64b integer division is very expensive. Github Issue #506 Co-authored-by: Vijay Krishnamoorthy <vijay_krish@apple.com>	2024-02-10 08:49:51 -08:00