-
Notifications
You must be signed in to change notification settings - Fork 26.3k
Add channels-last support to bundled_inputs #36764
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
Summary: This allows bundling inputs that are large uniform buffers in channels-last memory format. Test Plan: Unit test. [ghstack-poisoned]
💊 Build failures summary and remediationsAs of commit 05af798 (more details on the Dr. CI page):
🕵️ 2 new failures recognized by patternsThe following build failures do not appear to be due to upstream breakages:
|
Summary: This allows bundling inputs that are large uniform buffers in channels-last memory format. Test Plan: Unit test. Differential Revision: [D21142660](https://our.internmc.facebook.com/intern/diff/D21142660) [ghstack-poisoned]
| self.assertEqual(inflated[4][0].shape, (1 << 16,)) | ||
| self.assertAlmostEqual(inflated[4][0].mean().item(), 0, delta=0.025) | ||
| self.assertAlmostEqual(inflated[4][0].std().item(), 1, delta=0.02) | ||
| self.assertEqual(inflated[5][0].shape, (1 << 16,)) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Super nit: use -1 or len(samples)
Stack from ghstack:
Summary:
This allows bundling inputs that are large uniform buffers in
channels-last memory format.
Test Plan:
Unit test.
Differential Revision: D21142660