Skip to content

Add PyTorch autograd tests and fix interop/copy-back bugs #781

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.

Sign up for GitHub

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

Open
jhelferty-nv wants to merge 36 commits into
base: main
Choose a base branch
from
Open

Add PyTorch autograd tests and fix interop/copy-back bugs #781

Show file tree
Hide file tree
Changes from 8 commits
Commits
Show all changes
36 commits
Select commit Hold shift + click to select a range
c71e591
test using slangpy in pytorch sequence
jhelferty-nv Feb 5, 2026
bc3e4af
loss
jhelferty-nv Feb 5, 2026
ca43731
Activations
jhelferty-nv Feb 5, 2026
132608a
interleaving tests
jhelferty-nv Feb 5, 2026
6edd1b6
Fix PyTorch autograd compatibility by preventing spurious tensor vers…
jhelferty-nv Feb 5, 2026
6531182
activations with min/max
jhelferty-nv Feb 5, 2026
85bbe15
Strided slice with different ops
jhelferty-nv Feb 5, 2026
cae7267
minor tidy
jhelferty-nv Feb 5, 2026
f9b7910
more test coverage for copyback
jhelferty-nv Feb 9, 2026
b590ae3
Fix tensor writeback logic for both scalar broadcast and tensor param…
jhelferty-nv Feb 9, 2026
49eb4b9
Fix output copy-back for scalar broadcast functions
jhelferty-nv Feb 9, 2026
d5bd496
Clean up test docstrings and comments
jhelferty-nv Feb 9, 2026
719233b
Merge branch 'main' into add-pytorch-tests
jhelferty-nv Feb 9, 2026
e71a92b
Add explicit gradient copy-back tests
jhelferty-nv Feb 10, 2026
2cd616c
Refactor copy-back decision to cache time
jhelferty-nv Feb 11, 2026
293d4f5
Rename CachedOffsets to CachedBindingInfo
jhelferty-nv Feb 11, 2026
fe5ef48
Update comments to reflect copy-back decision logic
jhelferty-nv Feb 11, 2026
777fd6a
Add clarifying comment for copy-back flags in CachedBindingInfo
jhelferty-nv Feb 11, 2026
f10146b
Merge branch 'main' into add-pytorch-tests
jhelferty-nv Feb 11, 2026
bca0314
Fix renamed variable after merge with main
jhelferty-nv Feb 11, 2026
d8596fd
Fix gradient copy-back for raw torch.Tensor inputs on Vulkan/D3D12
jhelferty-nv Feb 11, 2026
0d31d28
Remove unused needs_grad_copyback field from CachedBindingInfo
jhelferty-nv Feb 11, 2026
eeea630
Condense copy-back comments to reduce duplication
jhelferty-nv Feb 11, 2026
9687116
Fix interop backward crash and broadcast stride zeroing
jhelferty-nv Feb 12, 2026
95d7081
Fix CUDA interop memory leak: free mapped buffer before destroying ex…
jhelferty-nv Feb 12, 2026
4b469c6
Add end-to-end workflow tests for slang-torch parity validation
jhelferty-nv Feb 12, 2026
b22bf96
Clarify e2e test docstrings: these are workflow pattern tests, not sl…
jhelferty-nv Feb 12, 2026
87a4314
Rename test_e2e_workflows.py to test_torch_autograd_workflows.py
jhelferty-nv Feb 12, 2026
8f9060d
Merge branch 'main' into add-pytorch-tests
jhelferty-nv Feb 12, 2026
d9dea92
Merge branch 'main' into add-pytorch-tests
jhelferty-nv Feb 17, 2026
4f1213e
Use Slang uniform type name for interop copy-back decisions
jhelferty-nv Feb 17, 2026
6930a6d
Use async memset for zeroed interop buffers
jhelferty-nv Feb 17, 2026
8b744a5
Use correct CUDA stream for interop buffer memset
jhelferty-nv Feb 19, 2026
01dedcd
Merge branch 'main' into add-pytorch-tests
jhelferty-nv Feb 19, 2026
e7f7c16
Merge branch 'main' into add-pytorch-tests
jhelferty-nv Feb 23, 2026
04e7a3d
Merge branch 'main' into add-pytorch-tests
jhelferty-nv Feb 25, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Loading
Loading