Skip to content

Torch input validation #149

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

Merged
vbharadwaj-bk merged 37 commits into from
Jul 9, 2025
Merged

Torch input validation #149

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
37 commits
Select commit Hold shift + click to select a range
a8979d5
add more kernel dims
Jun 8, 2025
f5dcf9a
simple test for torch-level shape checks
Jun 8, 2025
0277e33
add TORCH_CHECK to torch interface
Jun 8, 2025
6a0960e
Merge branch 'main' into shape-checking
Jun 10, 2025
4dfadc0
adds size checking, removes getters and replaces with const attributes.
Jun 11, 2025
f07b54f
adds kernel dims to jit via dict
Jun 11, 2025
9000d78
Merge remote-tracking branch 'origin' into shape-checking
Jun 11, 2025
a5c74d0
replace get_L3_dim with L3_dim
Jun 11, 2025
3bdfb3a
remove parenthesis
Jun 11, 2025
f56bff2
add a central location for type enums
Jun 13, 2025
029be2c
add location and type checking. expose LX_dim correctly. Pass types a…
Jun 13, 2025
715f7f9
pass additional info
Jun 13, 2025
39ed09a
better enum and proper transpose_perm size
Jun 14, 2025
46b37b7
better enum
Jun 14, 2025
af1ea30
better enum
Jun 14, 2025
2e41d4a
factor out executables
Jun 14, 2025
fe28a33
rename to shape_test
Jun 16, 2025
13514f6
add lazy fixtures so that I can easily parameterize tuples. lazy-fixt…
Jun 16, 2025
4247fc1
add enum to torch.dtype
Jun 16, 2025
9b80178
factor these into their own file for reuse
Jun 16, 2025
2ead2a6
added conftest for fixture reuse
Jun 16, 2025
119b8c7
factor out executables
Jun 16, 2025
1c85274
delete utils.py
Jun 16, 2025
4a85622
content moved elsewhere
Jun 17, 2025
8f7cbdc
moved stream test to torch_features level
Jun 17, 2025
cc6aae0
wip commit
Jun 23, 2025
109bc87
retain only core changes
Jul 4, 2025
efbf21d
Merge branch 'shape-checking' of github.com:PASSIONLab/OpenEquivarian…
Jul 4, 2025
1de1393
just forward direction testing, added pytest-subtests to to make test…
Jul 9, 2025
2869710
return double_backward_rawptrs which got removed accidentally
Jul 9, 2025
bd0734b
add other convs and add permutation where expected
Jul 9, 2025
5b801eb
Merge remote-tracking branch 'origin' into shape-checking
Jul 9, 2025
4cb1eeb
Some cleanup.
vbharadwaj-bk Jul 9, 2025
a2b610e
Cleanup.
vbharadwaj-bk Jul 9, 2025
f6fd4e3
Removed some fixtures that are only used once / reproducibility on ge…
vbharadwaj-bk Jul 9, 2025
a67b464
Light renaming.
vbharadwaj-bk Jul 9, 2025
8632844
Linted.
vbharadwaj-bk Jul 9, 2025
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