Skip to content
Draft
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions ext/TensorKitCUDAExt/truncation.jl
Original file line number Diff line number Diff line change
Expand Up @@ -74,3 +74,10 @@ function MatrixAlgebraKit._ind_intersect(A::CuVector{Bool}, B::CuVector{Int})
result[B] .= @view A[B]
return result
end

function MatrixAlgebraKit.truncation_error!(values::CuSectorVector, ind)
for (c, ind_c) in pairs(ind)
@. values[c] *= !ind_c
end
return norm(values)
end
Loading