[Executorch][llm] Fix flakyness of quantized sdpa test#15653
[Executorch][llm] Fix flakyness of quantized sdpa test#15653meta-codesync[bot] merged 2 commits intogh/kimishpatel/204/basefrom
Conversation
Without this fix when running diabled tets it fails with OMP: Error #131: Thread identifier invalid Differential Revision: [D86447540](https://our.internmc.facebook.com/intern/diff/D86447540/) [ghstack-poisoned]
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/15653
Note: Links to docs will display an error until the docs builds have been completed. ❌ 4 New FailuresAs of commit 6d81e96 with merge base aba44fd ( NEW FAILURES - The following jobs have failed:
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
This PR needs a
|
| self.q_shape = None | ||
| self.kv_shape = None | ||
| self.is_seq_at_dim_2 = True | ||
| # For some reason 4 threads doesnt work |
There was a problem hiding this comment.
Is there a concern with users choose a thread count that doesn't work?
There was a problem hiding this comment.
I think this is specifically related to OMP which I am definitely a bit surprised why it occurs here. This hasnt been observed on-device
Without this fix when running diabled tets it fails with OMP: Error #131: Thread identifier invalid Differential Revision: [D86447540](https://our.internmc.facebook.com/intern/diff/D86447540/) [ghstack-poisoned]
43cf63b
into
gh/kimishpatel/204/base
Stack from ghstack (oldest at bottom):
Without this fix when running diabled tets it fails with
OMP: Error #131: Thread identifier invalid
Differential Revision: D86447540