Skip to content

Commit aafbb9c

Browse files
Katharine Hyattkshyatt
authored andcommitted
Only use the scalar method for AMDGPU
1 parent 4bcb679 commit aafbb9c

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

src/implementations/truncation.jl

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,10 +17,7 @@ function truncate(::typeof(left_null!), (U, S), strategy::TruncationStrategy)
1717
# TODO: avoid allocation?
1818
extended_S = vcat(diagview(S), zeros(eltype(S), max(0, size(S, 1) - size(S, 2))))
1919
ind = findtruncated(extended_S, strategy)
20-
trunc_cols = collect(1:size(U, 2))[ind]
21-
Utrunc = similar(U, (size(U, 1), length(trunc_cols)))
22-
Utrunc .= U[:, trunc_cols]
23-
return Utrunc, ind
20+
return U[:, ind], ind
2421
end
2522
function truncate(::typeof(right_null!), (S, Vᴴ), strategy::TruncationStrategy)
2623
# TODO: avoid allocation?

0 commit comments

Comments
 (0)