Skip to content

Conversation

@yf225
Copy link
Contributor

@yf225 yf225 commented Dec 5, 2025

Suggested by @v0i0 in #1210 (comment):

how do i also specialize on the stride of a dim being 1? or whatever triton needs to know to generate vectorized loads

@yf225 yf225 requested review from jansel, oulgen and v0i0 December 5, 2025 19:36
@yf225 yf225 changed the title Allow using hl.specialize to specialize on tensor strides. Allow using hl.specialize to specialize on tensor strides Dec 5, 2025
@meta-cla meta-cla bot added the CLA Signed This label is managed by the Meta Open Source bot. label Dec 5, 2025
@yf225 yf225 force-pushed the specialize_on_stride branch 2 times, most recently from e8c62c7 to 99c1d79 Compare December 5, 2025 19:49
@yf225 yf225 requested a review from v0i0 December 5, 2025 19:50
@yf225 yf225 force-pushed the specialize_on_stride branch 2 times, most recently from a0f4845 to 27bd245 Compare December 5, 2025 20:13
@yf225 yf225 marked this pull request as draft December 5, 2025 20:28
@yf225 yf225 changed the title Allow using hl.specialize to specialize on tensor strides [wip] Allow using hl.specialize to specialize on tensor strides Dec 5, 2025
@yf225 yf225 force-pushed the specialize_on_stride branch 2 times, most recently from de9d90f to bc67e76 Compare December 5, 2025 21:35
@yf225 yf225 force-pushed the specialize_on_stride branch 3 times, most recently from 48e641f to 6138675 Compare December 5, 2025 22:27
@yf225 yf225 force-pushed the specialize_on_stride branch 6 times, most recently from 5968295 to 0e4cb60 Compare December 6, 2025 01:35
@yf225 yf225 force-pushed the specialize_on_stride branch from 0e4cb60 to 047d3f9 Compare December 6, 2025 01:45
@yf225 yf225 changed the title [wip] Allow using hl.specialize to specialize on tensor strides Allow using hl.specialize to specialize on tensor strides Dec 6, 2025
@yf225 yf225 marked this pull request as ready for review December 6, 2025 01:45
@yf225 yf225 merged commit 28cc903 into main Dec 12, 2025
16 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Meta Open Source bot.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants