Skip to content
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

Better slicing #1059

Merged
merged 19 commits into from
Apr 28, 2022
Merged

Better slicing #1059

merged 19 commits into from
Apr 28, 2022

Conversation

rdragos
Copy link
Contributor

@rdragos rdragos commented Apr 20, 2022

Closes #1064
Does a few things:

  1. (almost) complete kernel coverage for slicing op. We still need to do slicing for BitArray. See BitArray slicing #1066
  2. edsl support for slicing
  3. fills in some missing kernels for ConstantOp.

@rdragos rdragos changed the title [WIP] Better slicing Better slicing Apr 26, 2022
moose/src/logical/ops.rs Outdated Show resolved Hide resolved
Comment on lines +76 to +77
(HostPlacement, (Shape) -> Shape => [concrete] Self::logical_host_shape),
(HostPlacement, (Tensor) -> Tensor => [concrete] Self::logical_host_kernel),
Copy link
Member

Choose a reason for hiding this comment

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

nit: we often put the the logical kernel at the top for each placement (e.g identity)

Copy link
Contributor Author

Choose a reason for hiding this comment

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

I was thinking about this and how we should sort kernel types, I thought it would be good if we sort them by kernel types (whether they're runtime / replicated protocols / lowering) and then we sort everything alphabetically after the input types, something similar as @jvmncs did in a previous PR. I couldn't find something better for now.

Copy link
Member

@yanndupis yanndupis left a comment

Choose a reason for hiding this comment

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

Nice! Good work @rdragos !

@rdragos rdragos merged commit 6b6b89e into main Apr 28, 2022
@rdragos rdragos deleted the dragos/index-op branch April 28, 2022 16:33
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.

Extend slice operation kernel with existing type matrix
3 participants