Skip to content

Preserve factor dtype when transducting TT tensors - #39

Open
AryanRG wants to merge 1 commit into
tensorly:mainfrom
AryanRG:fix-tt-transduction-dtype
Open

AryanRG wants to merge 1 commit into
tensorly:mainfrom
AryanRG:fix-tt-transduction-dtype

Conversation

@AryanRG

@AryanRG AryanRG commented Oct 29, 2026

Copy link
Copy Markdown

After converting a TT tensor to float64, transduct() inserts a float32 core, causing to_tensor() to fail with a dtype mismatch. Allocate the new core from an existing factor so it inherits the factor's dtype and device.

Regression tests cover float32/float64 insertion, custom factors, reconstructed values, and state-dict round trips. The CPU full suite completes with Tucker convolution/TRL failures that also reproduce on the unpatched base. Documentation and source/wheel builds pass.

This branch has not been deployed

No deployments
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

Successfully merging this pull request may close these issues.

1 participant