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

#12527: refactor slice #12791

Merged
merged 1 commit into from
Sep 20, 2024
Merged

#12527: refactor slice #12791

merged 1 commit into from
Sep 20, 2024

Conversation

sjameelTT
Copy link
Contributor

@sjameelTT sjameelTT commented Sep 17, 2024

-use exclusive ends and move all the list splicing features to ttnn.slice
-add support for negative starts
-add negative starts and ends support to ttnn.slice
-move most pre-processing to C++
-refactor C++ code
-remove skip on bert ttnn.slice unit tests

Ticket

Link to Github Issue

Problem description

ttnn.slice and the tensor[..start:end:step] format had differences in feature support. Refactor slice to improve coverage.

What's changed

Describe the approach used to solve the problem.
Summarize the changes made and its impact.

Checklist

  • Post commit CI passes
  • Blackhole Post commit (if applicable)
  • Model regression CI testing passes (if applicable)
  • Device performance regression CI testing passes (if applicable)
  • New/Existing tests provide coverage for changes

 -use exclusive ends and move all the list splicing-exclusive features to ttnn.slice
 -use exclusive ends for ttnn.unpad
 -add support for negative starts
 -add negative starts and ends support to ttnn.slice
 -move most pre-processing to C++
 -refactor C++ code
 -remove skip on bert ttnn.slice unit tests
 -support 1D, 2D, 3D inputs on ttnn.slice
@sjameelTT sjameelTT merged commit 047cdd9 into main Sep 20, 2024
6 checks passed
@sjameelTT sjameelTT deleted the sjameel/slice_refactor branch September 20, 2024 21:02
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.