Skip to content

Support scheduling transforms of size-0 IterDomains#367

Merged
jacobhinkle merged 5 commits intomainfrom
split_merge_size0
May 18, 2023
Merged

Support scheduling transforms of size-0 IterDomains#367
jacobhinkle merged 5 commits intomainfrom
split_merge_size0

Conversation

@jacobhinkle
Copy link
Collaborator

Fixes #366. Removes checks that extents are non-zero when performing splits and merges. Note that this is only relevant for scheduling currently, since even for dynamic reshapes, their concretizations are performed with static sizes, where extent checks happen.

Kept the old one which exercises the pointwise scheduler too
@jacobhinkle
Copy link
Collaborator Author

@jjsjann123 these checks were introduced in this pytorch/pytorch@6ab2a6f. It seems safe to me to remove it but I wanted to check with you first. If there are some other concerns around size-0, I'd be happy to add new test cases for those.

@jacobhinkle jacobhinkle marked this pull request as ready for review May 18, 2023 14:34
@jacobhinkle jacobhinkle requested a review from jjsjann123 May 18, 2023 14:34
jacobhinkle added a commit that referenced this pull request May 18, 2023
Copy link
Collaborator

@jjsjann123 jjsjann123 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

:shipit:

I'm pretty sure this was just there to avoid some segfault from the old codegen time. Sorry for the confusion.

@jacobhinkle jacobhinkle merged commit 4493f63 into main May 18, 2023
@jacobhinkle jacobhinkle deleted the split_merge_size0 branch May 18, 2023 19:32
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Split/merge of size-0 slices

2 participants