xla
b205042b - Lower _linalg_svd (#4537)

Commit
2 years ago
Lower _linalg_svd (#4537) * Lower _linalg_svd This change lowers _linalg_svd by mostly reusing the existing svd lowering. However, there are a couple of notable differences between these two ops: 1. torch.svd()'s `some` is the opposite of torch.linalg.svd()'s `full_matrices`. Note that default value for both is True, so the default behavior is effectively the opposite. 2. torch.svd() returns V, whereas torch.linalg.svd() returns Vh (V transposed). 3. If `compute_uv` is False, torch.svd() returns zero-filled tensors for U and Vh, whereas torch.linalg.svd() returns empty tensors. * Run linter * Disable some more tests * Add comments to the disabled tests
Author
Parents
Loading