fix a set union bug (#2827)
Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/2827 It's a bug, we need to keep union the original set with the new set from the for loop, however the original code just overwrite the original set. Test with coreml + llama code ``` python3 -m examples.models.llama2.export_llama --coreml --use_kv_cache ``` Reviewed By: angelayi Differential Revision: D55680296 fbshipit-source-id: 906d508a276a91bd4b1caa295ce547e9686612ad
C
Chen Lai committed
a25dea6308700023201bbff426d38d300c459408
Parent: 51d3389
Committed by Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
on 4/3/2024, 7:06:20 PM