@@ -531,16 +531,22 @@ for V in spacelist
531531 #
532532 # TODO
533533 @timedtestset " Tensor product: test via norm preservation" begin
534- for T in (Float32, ComplexF64)
535- if UnitStyle (I) isa SimpleUnit || ! isempty (blocksectors (V2 ⊗ V1))
536- t1 = CUDA. rand (T, V2 ⊗ V3 ⊗ V1, V1 ⊗ V2)
537- t2 = CUDA. rand (T, V2 ⊗ V1 ⊗ V3, V1 ⊗ V1)
538- else
539- t1 = CUDA. rand (T, V3 ⊗ V4 ⊗ V5, (V1 ⊗ V2)' )
540- t2 = CUDA. rand (T, (V3 ⊗ V4 ⊗ V5)' , V1 ⊗ V2)
534+ for T in (ComplexF64, ) # Float32 case broken because of cuTENSOR
535+ @time " Construction" begin
536+ if UnitStyle (I) isa SimpleUnit || ! isempty (blocksectors (V2 ⊗ V1))
537+ t1 = CUDA. rand (T, V2 ⊗ V3 ⊗ V1, V1)
538+ t2 = CUDA. rand (T, V2 ⊗ V1 ⊗ V3, V1)
539+ else
540+ t1 = CUDA. rand (T, V3 ⊗ V4 ⊗ V5, V1' )
541+ t2 = CUDA. rand (T, (V3 ⊗ V4 ⊗ V5)' , V1)
542+ end
543+ end
544+ @time " Product" begin
545+ t = @constinferred (t1 ⊗ t2)
546+ end
547+ @time " Norm" begin
548+ @test norm (t) ≈ norm (t1) * norm (t2)
541549 end
542- t = @constinferred (t1 ⊗ t2)
543- @test norm (t) ≈ norm (t1) * norm (t2)
544550 end
545551 end
546552 symmetricbraiding && @timedtestset " Tensor product: test via conversion" begin
0 commit comments