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

Add unit tests for tensor interrogation #239

Open
wants to merge 9 commits into
base: main
Choose a base branch
from

Conversation

jwallwork23
Copy link
Contributor

@jwallwork23 jwallwork23 commented Jan 14, 2025

Closes #219.
Closes #211.

This PR adds unit tests for the get_rank and get_shape methods of torch_tensor, as well as for the torch_tensor_get_device_index util on both CPU and GPU.

I decided not to add a unit test for torch_tensor_to_array because it's already covered elsewhere (e.g., in the constructors tests).

@jwallwork23 jwallwork23 added autograd Tasks towards the online training / automatic differentiation feature testing Related to FTorch testing labels Jan 14, 2025
@jwallwork23 jwallwork23 self-assigned this Jan 14, 2025
@jwallwork23 jwallwork23 force-pushed the 219_unit-test-interrogate branch from d6c03c6 to 602d8f1 Compare January 15, 2025 11:14
@jwallwork23 jwallwork23 marked this pull request as ready for review January 15, 2025 12:01
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
autograd Tasks towards the online training / automatic differentiation feature testing Related to FTorch testing
Projects
None yet
1 participant