Jagrit Digani 1a48713d32 Update gather and scatter to not use Argument Encoder (#683) ...

* Replace argument encoder usage for gather and scatter

* Use constant address space for shapes and strides

* Split gather and scatter to improve compile times

* Enable the GPU tests

* Update the CI config

* Fix scatter dispatch for scalar indices

* Remove arg encoder utils

---------

Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>

2024-02-14 13:42:13 -08:00

..

config.yml

Update gather and scatter to not use Argument Encoder (#683)

2024-02-14 13:42:13 -08:00