Skip to content

Commit

Permalink
format
Browse files Browse the repository at this point in the history
  • Loading branch information
kmp5VT committed Nov 14, 2023
1 parent 9d020ce commit b1ec109
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions NDTensors/test/diag.jl
Original file line number Diff line number Diff line change
Expand Up @@ -52,11 +52,11 @@ using GPUArraysCore: @allowscalar
S2 = contract(dense(S), (2, -1), copy(V), (3, 4, -1))
@test @allowscalar S1 S2

S1M = dev(Matrix{elt}(undef, (2,12)))
S1M = dev(Matrix{elt}(undef, (2, 12)))
SM = array(S)
VM = transpose(reshape(array(V), (12,2)))
VM = transpose(reshape(array(V), (12, 2)))
NDTensors.mul!!(S1M, SM, VM, 1.0, 0.0)
@test @allowscalar reshape(S1M, (2,3,4)) array(S1)
@test @allowscalar reshape(S1M, (2, 3, 4)) array(S1)
end
end
@testset "DiagTensor contractions" begin
Expand Down

0 comments on commit b1ec109

Please sign in to comment.