Add index.Tensor and aten.logical_not (#3221)
Summary: Add missing llama ops for MPS delegate: - `index.Tensor` - `logical_not` `index.put` works correctly for generating 1 token, but gives incorrect results on 2nd token. This remains disabled. Summary of changes: - Adds missing llama2 ops - Adds support for launching Metal kernels instead of MPSGraph ops (if MPSGraph doesn't have the support) cc cccclai , shoumikhin Pull Request resolved: https://github.com/pytorch/executorch/pull/3221 Reviewed By: shoumikhin Differential Revision: D56447710 Pulled By: cccclai fbshipit-source-id: 778a485df5e67d1afd006b42f07b69c8a3961223
D
Denis Vieriu committed
02a6b66c2cfea96787e41aba8b12f343bd970322
Parent: e9d7868
Committed by Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
on 4/24/2024, 1:21:53 AM