-
Notifications
You must be signed in to change notification settings - Fork 233
feat: Introduce StridedLayout, support wrapping external allocations in Buffer, add StridedMemoryView.from_buffer #1283
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
Merged
leofang
merged 34 commits into
NVIDIA:main
from
stiepan:introduce_strided_layout_memview
Dec 10, 2025
Merged
Changes from 1 commit
Commits
Show all changes
34 commits
Select commit
Hold shift + click to select a range
85d4a3a
Add StridedLayout
stiepan 23b35fe
Support wrapping ptr in Buffer, create SMV from buffer and layout, dl…
stiepan 2470617
Documentation, linting, minor fixes
stiepan cf7eff5
Add NotImplemented copy_from/copy_to
stiepan 79010b8
Adjust flattening scalars to numpy/cupy behavior, fix shape validatio…
stiepan 4ca3567
Add StridedLayout tests
stiepan acdd6f8
Use explicit int32_t instead of int in integer fused type
stiepan 60a0d66
Disable (for now) exporting the SMV via dlpack
stiepan 1fa43d4
Revert dlpack changes
stiepan 67c6c5e
Support layouts up to 64 dims
stiepan a96bec5
Use cydriver to query memory attributes, fix managed memory handling,…
stiepan 91387b0
Test owner and mr cannot be specified together
stiepan 91c0af9
Test Buffer.close with owner
stiepan b74ef2c
Add envelope checks (rquires_size_in_bytes, offset_bounds)
stiepan 2c0343f
Docs, annotation fixes, remove dlpack export mentions
stiepan 598a2f1
Add SMV.from_buffer/view tests
stiepan bbb227b
Layout tests for SMV created from CAI
stiepan 26dfe3b
Fix missing host unregister call in buffer test
stiepan 3adae5c
Fix num attrib on re-try
stiepan 7554164
Call int on the buffer.handle
stiepan 68b7a79
Merge branch 'main' into introduce_strided_layout_memview
stiepan edace66
Don't enforce Buffer having an owner when creating SMV
stiepan 9f86322
Use np._s instead of a custom helper in the tests
stiepan 4335b2e
Take lanes into account when computing the itemsize
stiepan 6568e27
Merge branch 'main' into introduce_strided_layout_memview
stiepan cbf1d17
Move layout validation out of get_data_ptr helper
stiepan 4767fbb
Disambiguate all_axes mask for layout flattening, add range flattenin…
stiepan 5765a22
Bring back the intptr_t in SMV
stiepan 7ec8961
Merge branch 'main' into introduce_strided_layout_memview
stiepan db75aa0
Reorder methods, adjust SMV tests to from_dlpack/form_cai methods
stiepan 639ee5f
Move the Device import to top-level imports
stiepan 9fb5dfb
Merge branch 'main' into introduce_strided_layout_memview
stiepan 3375b4d
Merge branch 'main' into introduce_strided_layout_memview
stiepan 66fc6e8
Merge branch 'main' into introduce_strided_layout_memview
leofang File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Some comments aren't visible on the classic Files Changed page.
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Uh oh!
There was an error while loading. Please reload this page.