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

Generate CP ALS Runtime Comparisons Across Tensor Sizes and Dimensions #3

Open
ntjohnson1 opened this issue Dec 30, 2023 · 0 comments

Comments

@ntjohnson1
Copy link
Owner

ntjohnson1 commented Dec 30, 2023

Right now small tensors (in total entries and number of dimensions) appear to be faster in pyttb. However, for larger ones rusty_tensor (via PyO3 bindings) seems to win. We should figure out a range of sizes to compare against to look at the scaling trend. We should probably also investigate instrumenting rusty_tensor to see where we are spending the most time. I know we are doing a few unnecessary copies (at the rust/python boundary and at struct creation), but that might be a trivial amount of time for the scale of problems for now.

A couple unscientific comparisons on my machine:

Pyttb time: 3.5205395221710205
Rust time: 2.277113914489746
Shape: (100, 100, 100, 10)    
Pyttb time: 30.862141132354736
Rust time: 24.287920236587524
Shape: (100, 100, 100, 100)
@ntjohnson1 ntjohnson1 changed the title Generate Runtime Comparisons Across Tensor Sizes and Dimensions Generate CP ALS Runtime Comparisons Across Tensor Sizes and Dimensions Dec 30, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant