Mercurial > repos > public > sbplib_julia

comparison test/LazyTensors/tuple_manipulation_test.jl @ 1228:73f262a0a384 refactor/LazyTensors/tuple_manipulation

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add @inline to definition of split_tuple
author Jonatan Werpers <jonatan@werpers.com>
date Sun, 19 Feb 2023 11:41:40 +0100
parents 06b983f0d236
children 8f4259fbd39c
comparison
equal deleted inserted replaced
1227:06b983f0d236 1228:73f262a0a384
32 @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((1,), (2,)) 32 @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((1,), (2,))
33 @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((1,2), (1,2)) 33 @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((1,2), (1,2))
34 @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((1,2), (1)) 34 @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((1,2), (1))
35 35
36 36
37 split_tuple_static(t, ::Val{SZS}) where {SZS} = @inline LazyTensors.split_tuple(t,SZS) 37 split_tuple_static(t, ::Val{SZS}) where {SZS} = LazyTensors.split_tuple(t,SZS)
38 38
39 @inferred split_tuple_static((1,2,3,4,5,6), Val((3,1,2))) 39 @inferred split_tuple_static((1,2,3,4,5,6), Val((3,1,2)))
40
41 @inferred split_tuple_static((1,2,3,4),Val((3,1))) 40 @inferred split_tuple_static((1,2,3,4),Val((3,1)))
42 @inferred split_tuple_static((1,2,true,4),Val((3,1))) 41 @inferred split_tuple_static((1,2,true,4),Val((3,1)))
43 @inferred split_tuple_static((1,2,3,4,5,6),Val((3,2,1))) 42 @inferred split_tuple_static((1,2,3,4,5,6),Val((3,2,1)))
44 @inferred split_tuple_static((1,true,3),Val((1,1,1))) 43 @inferred split_tuple_static((1,true,3),Val((1,1,1)))
45 end 44 end