Motivation
Before going into the details of Julia's type system, we will spend a few minutes motivating the roles of a type system, which are:
- Structuring code
- Communicating to the compiler how a type will be used
The first aspect is important for the convenience of the programmer and enables abstractions in the language, the latter aspect is important for the speed of the generated code. Writing efficient Julia code is best viewed as a dialogue between the programmer and the compiler. [1]
Type systems according to Wikipedia:
- In computer science and computer programming, a data type or simply type is an attribute of data which tells the compiler or interpreter how the programmer intends to use the data.
- A type system is a logical system comprising a set of rules that assigns a property called a type to the various constructs of a computer program, such as variables, expressions, functions or modules. These types formalize and enforce the otherwise implicit categories the programmer uses for algebraic data types, data structures, or other components.
Good short answer to Static vs. Dynamic types
Structuring the code / enforcing the categories
The role of structuring the code and imposing semantic restriction means that the type system allows you to logically divide your program, and to prevent certain types of errors. Consider for example two types, Wolf
and Sheep
which share the same definition but the types have different names.
struct Wolf
name::String
energy::Int
end
struct Sheep
name::String
energy::Int
end
This allows us to define functions applicable only to the corresponding type
howl(wolf::Wolf) = println(wolf.name, " has howled.")
baa(sheep::Sheep) = println(sheep.name, " has baaed.")
Therefore the compiler (or interpreter) enforces that a wolf can only howl
and never baa
and vice versa a sheep can only baa
. In this sense, it ensures that howl(sheep)
and baa(wolf)
never happen.
baa(Sheep("Karl",3))
baa(Wolf("Karl",3))
Notice the type of error of the latter call baa(Wolf("Karl",3))
. Julia raises MethodError
which states that it has failed to find a function baa
for the type Wolf
(but there is a function baa
for type Sheep
).
For comparison, consider an alternative definition which does not have specified types
bark(animal) = println(animal.name, " has howled.")
baa(animal) = println(animal.name, " has baaed.")
in which case the burden of ensuring that a wolf will never baa rests upon the programmer which inevitably leads to errors (note that severely constrained type systems are difficult to use).
Intention of use and restrictions on compilers
Types play an important role in generating efficient code by a compiler, because they tells the compiler which operations are permitted, prohibited, and can indicate invariants of type (e.g. constant size of an array). If compiler knows that something is invariant (constant), it can exploit such information. As an example, consider the following two alternatives to represent a set of animals:
a = [Wolf("1", 1), Wolf("2", 2), Sheep("3", 3)]
b = (Wolf("1", 1), Wolf("2", 2), Sheep("3", 3))
where a
is an array which can contain arbitrary types and have arbitrary length whereas b
is a Tuple
which has fixed length in which the first two items are of type Wolf
and the third item is of type Sheep
. Moreover, consider a function which calculates the energy of all animals as
energy(animals) = mapreduce(x -> x.energy, +, animals)
A good compiler makes use of the information provided by the type system to generate efficient code which we can verify by inspecting the compiled code using @code_native
macro
@code_native debuginfo=:none energy(a)
@code_native debuginfo=:none energy(b)
one observes the second version produces more optimal code. Why is that?
- In the first representation,
a
, the animals are stored in anArray{Any}
which can have arbitrary size and can contain arbitrary animals. This means that the compiler has to compileenergy(a)
such that it works on such arrays. - In the second representation,
b
, the animals are stored in aTuple
, which specializes for lengths and types of items. This means that the compiler knows the number of animals and the type of each animal on each position within the tuple, which allows it to specialize.
This difference will indeed have an impact on the time of code execution. On my i5-8279U CPU, the difference (as measured by BenchmarkTools) is
using BenchmarkTools
@btime energy($(a))
@btime energy($(b))
70.2 ns (0 allocations: 0 bytes)
2.62 ns (0 allocations: 0 bytes)
Which nicely demonstrates that the choice of types affects performance. Does it mean that we should always use Tuples
instead of Arrays
? Surely not, it is just that each is better for different use-cases. Using Tuples means that the compiler will compile a special function for each length of tuple and each combination of types of items it contains, which is clearly wasteful.
Julia's type system
Julia is dynamicaly typed
Julia's type system is dynamic, which means that all types are resolved during runtime. But, if the compiler can infer types of all variables of the called function, it can specialize the function for that given type of variables which leads to efficient code. Consider a modified example where we represent two wolfpacks:
wolfpack_a = [Wolf("1", 1), Wolf("2", 2), Wolf("3", 3)]
wolfpack_b = Any[Wolf("1", 1), Wolf("2", 2), Wolf("3", 3)]
wolfpack_a
carries a type Vector{Wolf}
while wolfpack_b
has the type Vector{Any}
. This means that in the first case, the compiler knows that all items are of the type Wolf
and it can specialize functions using this information. In case of wolfpack_b
, it does not know which animal it will encounter (although all are of the same type), and therefore it needs to dynamically resolve the type of each item upon its use. This ultimately leads to less performant code.
@benchmark energy($(wolfpack_a))
@benchmark energy($(wolfpack_b))
3.7 ns (0 allocations: 0 bytes)
69.4 ns (0 allocations: 0 bytes)
To conclude, julia is indeed a dynamically typed language, but if the compiler can infer all types in a called function in advance, it does not have to perform the type resolution during execution, which produces performant code. This means and in hot (performance critical) parts of the code, you should be type stable, in other parts, it is not such big deal.
Classes of types
Julia divides types into three classes: primitive, composite, and abstract.
Primitive types
Citing the documentation: A primitive type is a concrete type whose data consists of plain old bits. Classic examples of primitive types are integers and floating-point values. Unlike most languages, Julia lets you declare your own primitive types, rather than providing only a fixed set of built-in ones. In fact, the standard primitive types are all defined in the language itself.
The definition of primitive types look as follows
primitive type Float16 <: AbstractFloat 16 end
primitive type Float32 <: AbstractFloat 32 end
primitive type Float64 <: AbstractFloat 64 end
and they are mainly used to jump-start julia's type system. It is rarely needed to define a special primitive type, as it makes sense only if you define special functions operating on its bits. This is almost exclusively used for exposing special operations provided by the underlying CPU / LLVM compiler. For example +
for Int32
is different from +
for Float32
as they call a different intrinsic operations. You can inspect this jump-starting of the type system yourself by looking at Julia's source.
julia> @which +(1,2)
+(x::T, y::T) where T<:Union{Int128, Int16, Int32, Int64, Int8, UInt128, UInt16, UInt32, UInt64, UInt8} in Base at int.jl:87
At int.jl:87
(+)(x::T, y::T) where {T<:BitInteger} = add_int(x, y)
we see that +
of integers is calling the function add_int(x, y)
, which is defined in the core part of the compiler in Intrinsics.cpp
(yes, in C++), exposed in Core.Intrinsics
From Julia docs: Core is the module that contains all identifiers considered "built in" to the language, i.e. part of the core language and not libraries. Every module implicitly specifies using Core, since you can't do anything without those definitions.
Primitive types are rarely used, and they will not be used in this course. We mention them for the sake of completeness and refer the reader to the official Documentation (and source code of Julia).
An example of use of primitive type is a definition of one-hot vector in the library PrimitiveOneHot
as
primitive type OneHot{K} <: AbstractOneHotArray{1} 32 end
where K
is the dimension of the one-hot vector.
Abstract types
An abstract type can be viewed as a set of concrete types. For example, an AbstractFloat
represents the set of concrete types (BigFloat,Float64,Float32,Float16)
. This is used mainly to define general methods for sets of types for which we expect the same behavior (recall the Julia design motivation: if it quacks like a duck, waddles like a duck and looks like a duck, chances are it's a duck). Abstract types are defined with abstract type TypeName end
. For example the following set of abstract types defines part of julia's number system.
abstract type Number end
abstract type Real <: Number end
abstract type Complex <: Number end
abstract type AbstractFloat <: Real end
abstract type Integer <: Real end
abstract type Signed <: Integer end
abstract type Unsigned <: Integer end
where <:
means "is a subtype of" and it is used in declarations where the right-hand is an immediate supertype of a given type (Integer
has the immediate supertype Real
.) If the supertype is not supplied, it is considered to be Any, therefore in the above definition Number
has the supertype Any
.
We can list childrens of an abstract type using function subtypes
using InteractiveUtils: subtypes # hide
subtypes(AbstractFloat)
and we can also list the immediate supertype
or climb the ladder all the way to Any
using supertypes
using InteractiveUtils: supertypes # hide
supertypes(AbstractFloat)
supertype
and subtypes
print only types defined in Modules that are currently loaded to your workspace. For example with Julia without any Modules, subtypes(Number)
returns [Complex, Real]
, whereas if I load Mods
package implementing numbers defined over finite field, the same call returns [Complex, Real, AbstractMod]
.
It is relatively simple to print a complete type hierarchy of
using AbstractTrees
function AbstractTrees.children(t::Type)
t === Function ? Vector{Type}() : filter!(x -> x !== Any,subtypes(t))
end
AbstractTrees.printnode(io::IO,t::Type) = print(io,t)
print_tree(Number)
The main role of abstract types allows is in function definitions. They allow to define functions that can be used on variables with types with a given abstract type as a supertype. For example we can define a sgn
function for all real numbers as
sgn(x::Real) = x > 0 ? 1 : x < 0 ? -1 : 0
and we know it would be correct for all real numbers. This means that if anyone creates a new subtype of Real
, the above function can be used. This also means that it is expected that comparison operations are defined for any real number. Also notice that Complex
numbers are excluded, since they do not have a total order.
For unsigned numbers, the sgn
can be simplified, as it is sufficient to verify if they are different (greater) than zero, therefore the function can read
sgn(x::Unsigned) = x > 0 ? 1 : 0
and again, it applies to all numbers derived from Unsigned
. Recall that Unsigned <: Integer <: Real,
how does Julia decide, which version of the function sgn
to use for UInt8(0)
? It chooses the most specific version, and thus for sgn(UInt8(0))
it will use sgn(x::Unsinged)
. If the compiler cannot decide, typically it encounters an ambiguity, it throws an error and recommends which function you should define to resolve it.
The above behavior allows to define default "fallback" implementations and while allowing to specialize for sub-types. A great example is matrix multiplication, which has a generic (and slow) implementation with many specializations, which can take advantage of structure (sparse, banded), or use optimized implementations (e.g. blas implementation for dense matrices with eltype Float32
and Float64
).
Posit numbers are an alternative to IEEE764 format to store floating points (Real
) numbers. They offer higher precision around zero and wider dynamic range of representable numbers. When julia performs matrix multiplication with Float32
/ Float64
, it will use BLAS routines written in C, which are very performant.
A = rand(Float32, 32, 32)
B = rand(Float32, 32, 16)
A * B
julia> @which A*B
*(A::Union{LinearAlgebra.Adjoint{<:Any, <:StridedMatrix{var"#s994"}}, LinearAlgebra.Transpose{<:Any, <:StridedMatrix{var"#s994"}}, StridedMatrix{var"#s994"}} where var"#s994"<:Union{Float32, Float64}, B::Union{LinearAlgebra.Adjoint{<:Any, <:StridedMatrix{var"#s128"}}, LinearAlgebra.Transpose{<:Any, <:StridedMatrix{var"#s128"}}, StridedMatrix{var"#s128"}} where var"#s128"<:Union{Float32, Float64})
@ LinearAlgebra ~/.julia/juliaup/julia-1.10.5+0.aarch64.apple.darwin14/share/julia/stdlib/v1.10/LinearAlgebra/src/matmul.jl:111
We can now convert both matrices to posit representation. We can still multiply them, but the function which will now perform the multiplication is a generic multiplication.
using SoftPosit
A = Posit32.(A)
B = Posit32.(B)
A * B
@which A*B
*(A::AbstractMatrix, B::AbstractMatrix)
@ LinearAlgebra ~/.julia/juliaup/julia-1.10.5+0.aarch64.apple.darwin14/share/julia/stdlib/v1.10/LinearAlgebra/src/matmul.jl:104
Again, Julia does not make a difference between abstract types defined in Base
libraries shipped with the language and those defined by you (the user). All are treated the same.
From Julia documentation: Abstract types cannot be instantiated, which means that we cannot create a variable that would have an abstract type (try typeof(Number(1f0))
). Also, abstract types cannot have any fields, therefore there is no composition (there are lengthy discussions of why this is so, one of the most definite arguments of creators is that abstract types with fields frequently lead to children types not using some fields (consider circle vs. ellipse)).
Composite types
Composite types are similar to struct
in C (they even have the same memory layout) as they logically join together other types. It is not a great idea to think about them as objects (in OOP sense), because objects tie together data and functions on owned data. Contrary in Julia (as in C), functions operate on data of structures, but are not tied to them and they are defined outside them. Composite types are workhorses of Julia's type system, as user-defined types are mostly composite (or abstract).
Composite types are defined using struct TypeName [fields] end
. To define a position of an animal on the Euclidean plane as a type, we would write
struct PositionF64
x::Float64
y::Float64
end
which defines a structure with two fields x
and y
of type Float64
. Julia's compiler creates a default constructor, where both (but generally all) arguments are converted using (convert(Float64, x), convert(Float64, y)
to the correct type. This means that we can construct a PositionF64 with numbers of different type that are convertable to Float64, e.g. PositionF64(1,1//2)
but we cannot construct PositionF64
where the fields would be of different type (e.g. Int
, Float32
, etc.) or they are not trivially convertable (e.g. String
).
Fields in composite types do not have to have a specified type. We can define a VaguePosition
without specifying the type
struct VaguePosition
x
y
end
This works as the definition above except that the arguments are not converted to Float64
now. One can store different values in x
and y
, for example String
(e.g. VaguePosition("Hello","world")). Although the above definition might be convenient, it limits the compiler's ability to specialize, as the type VaguePosition
does not carry information about type of x
and y
, which has a negative impact on the performance. Let's demonstrate it on an example of random walk. where we implement function move
which move position by position.
using BenchmarkTools
move(a,b) = typeof(a)(a.x+b.x, a.y+b.y)
δx = [PositionF64(rand(), rand()) for _ in 1:100]
x₀ = PositionF64(rand(), rand())
δy = [VaguePosition(rand(), rand()) for _ in 1:100]
y₀ = VaguePosition(rand(), rand())
@benchmark foldl(move, $(δx); init = $(x₀))
@benchmark foldl(move, $(δy); init = $(y₀))
Giving fields of a composite type an abstract type does not really solve the problem of the compiler not knowing the type. In this example, it still does not know, if it should use instructions for Float64
or Int8
. From the perspective of generating optimal code, the definition of LessVaguePosition
and VaguePosition
are equally uninformative to the compiler as it cannot assume anything about the code. However, the LessVaguePosition
will ensure that the position will contain only numbers, hence catching trivial errors like instantiating VaguePosition
with non-numeric types for which arithmetic operators will not be defined (recall the discussion on the beginning of the lecture).
struct LessVaguePosition
x::Real
y::Real
end
δz = [LessVaguePosition(rand(), rand()) for _ in 1:100]
z₀ = LessVaguePosition(rand(), rand())
@benchmark foldl(move, $(δz); init = $(z₀))
nothing #hide
All structs defined above are immutable (as we have seen above in the case of Tuple
), which means that one cannot change a field (unless the struct wraps a container, like and array, which allows that). For example this raises an error
a = LessVaguePosition(1,2)
a.x = 2
If one needs to make a struct mutable, use the keyword mutable
before the keyword struct
as
mutable struct MutablePosition
x::Float64
y::Float64
end
In mutable structures, we can change the values of fields.
a = MutablePosition(1e0, 2e0)
a.x = 2;
a
The functional difference between those is that you are not allowed to change fields of non-mutable structures. Therefore when the structure needs to be changed, you need to construct new structure, which means calling constructor, where you can enforce certain properties (through inner constructor).
There is also difference for the compiler. When a struct is non-mutable, the fields in memory stores the actual values, and therefore they can be stored on stack (this is possible only if all fields are of primitive types). The operations are fast, because there is no pointer dereferencing, there is no need to allocate memory (when they can be store on stack), and therefore less presure on Garbage Collector. When stack is mutable, the structure contains a pointer to a memory location on the heap, which contains the value. This means to access the value, we first need to read the pointer to the memory location and then read the value (two fetches from the memory). Moreover, the value has to be stored on Heap, which means that we need to allocate memory during construction, which is expensive.
The difference can be seen from
a, b = PositionF64(1,2), PositionF64(1,2)
@code_native debuginfo=:none move(a,b)
a, b = MutablePosition(1,2), MutablePosition(1,2)
@code_native debuginfo=:none move(a,b)
Why there is just one addition?
Also, the mutability is costly.
δx = [PositionF64(rand(), rand()) for _ in 1:100]
x₀ = PositionF64(rand(), rand())
δz = [MutablePosition(rand(), rand()) for _ in 1:100]
z₀ = MutablePosition(rand(), rand())
@benchmark foldl(move, $(δx); init = $(x₀))
@benchmark foldl(move, $(δz); init = $(z₀))
Parametric types
So far, we had to trade-off flexibility for generality in type definitions. We either have structured with a fixed type, which are fast, or we had structures with a general type information, but they are slow. Can we have both? The answer is affirmative. The way to achieve this flexibility in definitions of the type while being able to generate optimal code is to parametrize the type definition. This is achieved by replacing types with a parameter (typically a single uppercase character) and decorating in definition by specifying different type in curly brackets. For example
struct PositionT{T}
x::T
y::T
end
u64 = [PositionT(rand(), rand()) for _ in 1:100]
u32 = [PositionT(rand(Float32), rand(Float32)) for _ in 1:100]
@benchmark foldl(move, $(u64))
@benchmark foldl(move, $(u32))
Notice that the compiler can take advantage of specializing for different types (which does not have an effect here as in modern processors addition of Float
and Int
takes the same time).
v = [PositionT(Int8(rand(1:100)), Int8(rand(1:100))) for _ in 1:100];
@benchmark reduce(move, v)
nothing #hide
The above definition suffers the same problem as VaguePosition
, which is that it allows us to instantiate the PositionT
with non-numeric types, e.g. String
. We solve this by restricting the types T
to be children of some supertype, in this case Real
struct Position{T<:Real}
x::T
y::T
end
which will throw an error if we try to initialize it with Position("1.0", "2.0")
. Notice the flexibility we have achieved. We can use Position
to store (and later compute) not only over Float32
/ Float64
but any real numbers defined by other packages, for example with Posit
s.
using SoftPosit
p32 = [Position(Posit32(rand(Float32)), Posit32(rand(Float32))) for _ in 1:100];
@benchmark foldl(move, $(p32))
The above test with Posit
is slow, because there is no hardware support for their operations. Trying to construct the Position
with different type of real numbers will fail, example Position(1f0,1e0),
because through type definition we enforce them to have equal type.
Naturally, fields in structures can be of different types, as is in the below pointless example.
struct PositionXY{X<:Real, Y<:Real}
x::X
y::Y
end
The type can be parametrized by a concrete types. This is useful to communicate the compiler some useful informations, for example size of arrays.
struct PositionZ{T<:Real,Z}
x::T
y::T
end
PositionZ{Int64,1}(1,2)
Abstract parametric types
Like Composite types, Abstract types can also have parameters. These parameters define types that are common for all child types. A very good example is Julia's definition of arrays of arbitrary dimension N
and type T
of its items as
abstract type AbstractArray{T,N} end
Different T
and N
give rise to different variants of AbstractArrays
, therefore AbstractArray{Float32,2}
is different from AbstractArray{Float64,2}
and from AbstractArray{Float64,1}.
Note that these are still Abstract
types, which means you cannot instantiate them. Their purpose is
- to allow to define operations for broad class of concrete types
- to inform the compiler about constant values, which can be used
Notice in the above example that parameters of types do not have to be types, but can also be values of primitive types, as in the above example of AbstractArray
N
is the number of dimensions which is an integer value.
For convenience, it is common to give some important partially instantiated Abstract types an alias, for example AbstractVector
as
const AbstractVector{T} = AbstractArray{T,1}
is defined in array.jl:23
(in Julia 1.6.2), which allows us to define for example general prescription for the dot
product of two abstract vectors as
function dot(a::AbstractVector, b::AbstractVector)
@assert length(a) == length(b)
mapreduce(*, +, a, b)
end
You can verify that the above general function can be compiled to performant code if specialized for particular arguments.
using InteractiveUtils: @code_native
@code_native debuginfo=:none mapreduce(*,+, [1,2,3], [1,2,3])
More on the use of types in function definitions
Terminology
A function refers to a set of "methods" for a different combination of type parameters (the term function can be therefore considered as referring to a mere name). Methods define different behavior for different types of arguments for a given function. For example
move(a::Position, b::Position) = Position(a.x + b.x, a.y + b.y)
move(a::Vector{<:Position}, b::Vector{<:Position}) = move.(a,b)
move
refers to a function with methods move(a::Position, b::Position)
and move(a::Vector{<:Position}, b::Vector{<:Position})
. When different behavior on different types is defined by a programmer, as shown above, it is also called implementation specialization. There is another type of specialization, called compiler specialization, which occurs when the compiler generates different functions for you from a single method. For example for
move(Position(1,1), Position(2,2))
move(Position(1.0,1.0), Position(2.0,2.0))
move(Position(Posit8(1),Posit8(1)), Position(Posit8(2),Posit8(2)))
Frequent problems
- Why does the following fail?
foo(a::Vector{Real}) = println("Vector{Real}")
foo([1.0,2,3])
Julia's type system is invariant, which means that Vector{Real}
is different from Vector{Float64}
and from Vector{Float32}
, even though Float64
and Float32
are sub-types of Real
. Therefore typeof([1.0,2,3])
isa Vector{Float64}
which is not subtype of Vector{Real}.
For covariant languages, this would be true. For more information on variance in computer languages, see here. If the above definition of foo
should be applicable to all vectors which has elements of subtype of Real
we have define it as
foo(a::Vector{T}) where {T<:Real} = println("Vector{T} where {T<:Real}")
or equivalently but more tersely as
foo(a::Vector{<:Real}) = println("Vector{T} where {T<:Real}")
- Diagonal rule says that a repeated type in a method signature has to be a concrete type (this is to avoid ambiguity if the repeated type is used inside function definition to define a new variable to change type of variables). Consider for example the function below
move(a::T, b::T) where {T<:Position} = T(a.x + by.x, a.y + by.y)
we cannot call it with move(Position(1.0,2.0), Position(1,2))
, since in this case Position(1.0,2.0)
is of type Position{Float64}
while Position(1,2)
is of type Position{Int64}
.
The Diagonal rule applies to parametric types as well.
move(a::Position{T}, b::Position{T}) where {T} = T(a.x + by.x, a.y + by.y)
- When debugging why arguments do not match a particular method definition, it is useful to use
typeof
,isa
, and<:
commands. For example
typeof(Position(1.0,2.0))
typeof(Position(1,2))
Position(1,2) isa Position{Float64}
Position(1,2) isa Position{Real}
Position(1,2) isa Position{<:Real}
typeof(Position(1,2)) <: Position{<:Float64}
typeof(Position(1,2)) <: Position{<:Real}
Intermezzo: How does the Julia compiler work?
Let's walk through an example. Consider the following definitions
move(a::Position, by::Position) = Position(a.x + by.x, a.y + by.y)
move(a::T, by::T) where {T<:Position} = Position(a.x + by.x, a.y + by.y)
move(a::Position{Float64}, by::Position{Float64}) = Position(a.x + by.x, a.y + by.y)
move(a::Vector{<:Position}, by::Vector{<:Position}) = move.(a, by)
move(a::Vector{<:Position}, by::Position) = move.(a, by)
and a function call
a = Position(1.0, 1.0)
by = Position(2.0, 2.0)
move(a, by)
The compiler knows that you call the function
move
.The compiler infers the type of the arguments. You can view the result with
(typeof(a),typeof(by))
The compiler identifies all
move
-methods with arguments of type(Position{Float64}, Position{Float64})
:m = Base.method_instances(move, (typeof(a), typeof(by)), Base.get_world_counter())
a) If the method has been specialized (compiled), then the arguments are prepared and the method is invoked. The compiled specialization can be seen from
m.cache
b) If the method has not been specialized (compiled), the method is compiled for the given type of arguments and continues as in step 4a.
A compiled function is therefore a "blob" of native code living in a particular memory location. When Julia calls a function, it needs to pick the right block corresponding to a function with particular type of parameters.
If the compiler cannot narrow the types of arguments to concrete types, it has to perform the above procedure inside the called function, which has negative effects on performance, as the type resolution and identification of the methods can be slow, especially for methods with many arguments (e.g. 30ns for a method with one argument, 100 ns for method with two arguments). You always want to avoid run-time resolution inside the performant loop!!! Recall the above example
wolfpack_a = [Wolf("1", 1), Wolf("2", 2), Wolf("3", 3)]
wolfpack_b = Any[Wolf("1", 1), Wolf("2", 2), Wolf("3", 3)]
@benchmark energy($(wolfpack_a))
@benchmark energy($(wolfpack_b))
An interesting intermediate between fully abstract and fully concrete type happens, when the compiler knows that arguments have abstract type, which is composed of a small number of concrete types. This case called Union-Splitting, which happens when there is just a little bit of uncertainty. Julia will do something like
argtypes = typeof(args)
push!(execution_stack, args)
if T == Tuple{Int, Bool}
@goto compiled_blob_1234
else # the only other option is Tuple{Float64, Bool}
@goto compiled_blob_1236
end
For example
const WolfOrSheep = Union{Wolf, Sheep}
wolfpack_c = WolfOrSheep[Wolf("1", 1), Wolf("2", 2), Wolf("3", 3)]
@benchmark energy($(wolfpack_c))
Thanks to union splitting, Julia is able to have performant operations on arrays with undefined / missing values for example
[1, 2, 3, missing] |> typeof
More on matching methods and arguments
In the above process, the step, where Julia looks for a method instance with corresponding parameters can be very confusing. The rest of this lecture will focus on this. For those who want to have a formal background, we recommend talk of Francesco Zappa Nardelli and / or the one of Jan Vitek.
When Julia needs to specialize a method instance, it needs to find it among multiple definitions. A single function can have many method instances, see for example methods(+)
which lists all method instances of the +
-function. How does Julia select the proper one?
It finds all methods where the type of arguments match or are subtypes of restrictions on arguments in the method definition.
If there are multiple matches, the compiler selects the most specific definition.
If the compiler cannot decide, which method instance to choose, it throws an error.
confused_move(a::Position{Float64}, by) = Position(a.x + by.x, a.y + by.y)
confused_move(a, by::Position{Float64}) = Position(a.x + by.x, a.y + by.y)
confused_move(Position(1.0,2.0), Position(1.0,2.0))
- If it cannot find a suitable method, it throws an error.
move(Position(1,2), VaguePosition("hello","world"))
Some examples: Consider following definitions
move(a::Position, by::Position) = Position(a.x + by.x, a.y + by.y)
move(a::T, by::T) where {T<:Position} = T(a.x + by.x, a.y + by.y)
move(a::Position{Float64}, by::Position{Float64}) = Position(a.x + by.x, a.y + by.y)
move(a::Vector{<:Position}, by::Vector{<:Position}) = move.(a, by)
move(a::Vector{T}, by::Vector{T}) where {T<:Position} = move.(a, by)
move(a::Vector{<:Position}, by::Position) = move.(a, by)
Which method will compiler select for
move(Position(1.0,2.0), Position(1.0,2.0))
The first three methods match the types of arguments, but the compiler will select the third one, since it is the most specific.
Which method will compiler select for
move(Position(1,2), Position(1,2))
Again, the first and second method definitions match the argument, but the second is the most specific.
Which method will the compiler select for
move([Position(1,2)], [Position(1,2)])
Again, the fourth and fifth method definitions match the argument, but the fifth is the most specific.
move([Position(1,2), Position(1.0,2.0)], [Position(1,2), Position(1.0,2.0)])
A bizzare definition which you can encounter
The following definition of a one-hot matrix is taken from Flux.jl
struct OneHotArray{T<:Integer, L, N, var"N+1", I<:Union{T,AbstractArray{T, N}}} <: AbstractArray{Bool, var"N+1"}
indices::I
end
The parameters of the type carry information about the type used to encode the position of one
in each column in T
, the dimension of one-hot vectors in L
, the dimension of the storage of indices
in N
(which is zero for OneHotVector
and one for OneHotMatrix
), number of dimensions of the OneHotArray
in var"N+1"
and the type of underlying storage of indices I
.