Mercurial > repos > public > sbplib_julia

comparison test/testLazyTensors.jl @ 453:c1ae837f1a2e feature/inflated_tensormapping

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add some type inferense tests. Failing..
author Jonatan Werpers <jonatan@werpers.com>
date Tue, 20 Oct 2020 07:54:59 +0200
parents aeda2698166d
children b86312d14873
comparison
equal deleted inserted replaced
452:aeda2698166d 453:c1ae837f1a2e
304 I = IdentityMapping{Float64}((4,5)) 304 I = IdentityMapping{Float64}((4,5))
305 v = rand(4,5) 305 v = rand(4,5)
306 @inferred (I*v)[3,2] 306 @inferred (I*v)[3,2]
307 @inferred (I'*v)[3,2] 307 @inferred (I'*v)[3,2]
308 @inferred range_size(I) 308 @inferred range_size(I)
309
310 @inferred range_dim(I)
311 @inferred domain_dim(I)
309 end 312 end
310 313
311 @testset "InflatedTensorMapping" begin 314 @testset "InflatedTensorMapping" begin
312 I(sz...) = IdentityMapping(sz...) 315 I(sz...) = IdentityMapping(sz...)
313 316
339 v = rand(domain_size(tm)...) 342 v = rand(domain_size(tm)...)
340 343
341 @tullio IAIv[a,b,c,d] := Ã[c,i]*v[a,b,i,d] 344 @tullio IAIv[a,b,c,d] := Ã[c,i]*v[a,b,i,d]
342 @test tm*v ≈ IAIv rtol=1e-14 345 @test tm*v ≈ IAIv rtol=1e-14
343 346
344 end 347 @inferred LazyTensors.split_index(tm,1,1,1,1)
345 348 @inferred (tm*v)[1,1,1,1]
346 end 349
350 end
351
352 end