Implements GraphMemoryResource#1235

Merged

Andy-Jost merged 24 commits intoNVIDIA:mainfrom

Andy-Jost:dmr-graph-support-2

Nov 24, 2025

Commits on Nov 12, 2025

Implement non-pooling memory allocation.
Andy-Jost
committed
Add GraphMemoryResource.
Andy-Jost
committed
Remove mempool_enabled option now that GraphMemoryResource is ready.
Andy-Jost
committed
Add docstring and make GraphMemoryResource a singleton.
Andy-Jost
committed
Move tests to a separate file.
Andy-Jost
committed
Add errors for DeviceMemoryResource and GraphMemoryResource when graph capture state is not as expected.
Andy-Jost
committed
Add tests for attributes and memory allocation escaping graphs.
Andy-Jost
committed
Simplify logic for converting IsStreamT arguments.
Andy-Jost
committed
Standardize Stream arguments to IsStreamT. Update Buffer and MemoryResource methods to take any kind of stream-providing object. Update graph allocation tests.
Andy-Jost
committed
Add tests for IsStreamT conversions.
Andy-Jost
committed
Expand files named _gmr.*. Add __eq__ and __hash__ support to StreamWrapper (testing only)
Andy-Jost
committed
Fix format/lint issues.
Andy-Jost
committed

Commits on Nov 13, 2025

Minor clean up.
Andy-Jost
committed

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Implements GraphMemoryResource#1235

Implements GraphMemoryResource#1235
Andy-Jost merged 24 commits intoNVIDIA:mainfrom
Andy-Jost:dmr-graph-support-2

Commits on Nov 12, 2025

Implement non-pooling memory allocation.

Add GraphMemoryResource.

Remove mempool_enabled option now that GraphMemoryResource is ready.

Add docstring and make GraphMemoryResource a singleton.

Move tests to a separate file.

Add errors for DeviceMemoryResource and GraphMemoryResource when graph capture state is not as expected.

Add tests for attributes and memory allocation escaping graphs.

Simplify logic for converting IsStreamT arguments.

Standardize Stream arguments to IsStreamT. Update Buffer and MemoryResource methods to take any kind of stream-providing object. Update graph allocation tests.

Add tests for IsStreamT conversions.

Expand files named _gmr.*. Add eq and hash support to StreamWrapper (testing only)

Fix format/lint issues.

Commits on Nov 13, 2025

Minor clean up.

Commits on Nov 20, 2025

Change public signatures to accept "Stream | GraphBuiler" where only streams were accepted. Add a helper Stream_accept to accept Stream-like arguments throughout. Revert changes to Stream._init tha…

Add deprecation warning when stream protocol is used with launch.

Merge branch 'main' into dmr-graph-support-2

Fix builds post-merge.

Simplify GraphMemoryResourceAttributes.

Simplify Stream_accept. Default arguments can more easily be handled outside that function.

Adjust tests for platform-dependent behavior.

Disable additional tests for platform-dependent behavior.

Merge branch 'main' into dmr-graph-support-2

Commits on Nov 22, 2025

Adjust deallocation stream for legacy memory resources to avoid platform-dependent errors. Add dependence on mempool_device where needed for certain tests. Touch-ups.

Adjust test_graph_alloc to launch the graph multiple times.

Implements GraphMemoryResource#1235Andy-Jost merged 24 commits intoNVIDIA:mainNVIDIA/cuda-python:mainfrom Andy-Jost:dmr-graph-support-2Andy-Jost/cuda-python:dmr-graph-support-2Copy head branch name to clipboard

Commits

Commits on Nov 12, 2025

Commits on Nov 13, 2025

Commits on Nov 20, 2025

Commits on Nov 22, 2025

Implements GraphMemoryResource#1235
Andy-Jost merged 24 commits intoNVIDIA:mainfrom
Andy-Jost:dmr-graph-support-2