Performance Tips

In the following sections, we briefly go through a few techniques that can help make your Julia code run as fast as possible.

Avoid global variables

A global variable might have its value, and therefore its type, change at any point. This makes it difficult for the compiler to optimize code using global variables. Variables should be local, or passed as arguments to functions, whenever possible.

Any code that is performance critical or being benchmarked should be inside a function.

We find that global names are frequently constants, and declaring them as such greatly improves performance:

1. const DEFAULT_VAL = 0

Uses of non-constant globals can be optimized by annotating their types at the point of use:

1. global x = rand(1000)

3. function loop_over_global()
4. s = 0.0
5. for i in x::Vector{Float64}
6. s += i
7. end
8. return s
9. end

Passing arguments to functions is better style. It leads to more reusable code and clarifies what the inputs and outputs are.

Note

All code in the REPL is evaluated in global scope, so a variable defined and assigned at top level will be a global variable. Variables defined at top level scope inside modules are also global.

In the following REPL session:

1. julia> x = 1.0

is equivalent to:

1. julia> global x = 1.0

so all the performance issues discussed previously apply.

Measure performance with @time and pay attention to memory allocation

A useful tool for measuring performance is the @time macro. We here repeat the example with the global variable above, but this time with the type annotation removed:

1. julia> x = rand(1000);

3. julia> function sum_global()
4. s = 0.0
5. for i in x
6. s += i
7. end
8. return s
9. end;

11. julia> @time sum_global()
12. 0.009639 seconds (7.36 k allocations: 300.310 KiB, 98.32% compilation time)
13. 496.84883432553846

15. julia> @time sum_global()
16. 0.000140 seconds (3.49 k allocations: 70.313 KiB)
17. 496.84883432553846

On the first call (@time sum_global()) the function gets compiled. (If you’ve not yet used @time in this session, it will also compile functions needed for timing.) You should not take the results of this run seriously. For the second run, note that in addition to reporting the time, it also indicated that a significant amount of memory was allocated. We are here just computing a sum over all elements in a vector of 64-bit floats so there should be no need to allocate memory (at least not on the heap which is what @time reports).

Unexpected memory allocation is almost always a sign of some problem with your code, usually a problem with type-stability or creating many small temporary arrays. Consequently, in addition to the allocation itself, it’s very likely that the code generated for your function is far from optimal. Take such indications seriously and follow the advice below.

If we instead pass x as an argument to the function it no longer allocates memory (the allocation reported below is due to running the @time macro in global scope) and is significantly faster after the first call:

1. julia> x = rand(1000);

3. julia> function sum_arg(x)
4. s = 0.0
5. for i in x
6. s += i
7. end
8. return s
9. end;

11. julia> @time sum_arg(x)
12. 0.006202 seconds (4.18 k allocations: 217.860 KiB, 99.72% compilation time)
13. 496.84883432553846

15. julia> @time sum_arg(x)
16. 0.000005 seconds (1 allocation: 16 bytes)
17. 496.84883432553846

The 1 allocation seen is from running the @time macro itself in global scope. If we instead run the timing in a function, we can see that indeed no allocations are performed:

1. julia> time_sum(x) = @time sum_arg(x);

3. julia> time_sum(x)
4. 0.000001 seconds
5. 496.84883432553846

In some situations, your function may need to allocate memory as part of its operation, and this can complicate the simple picture above. In such cases, consider using one of the tools below to diagnose problems, or write a version of your function that separates allocation from its algorithmic aspects (see Pre-allocating outputs).

Note

For more serious benchmarking, consider the BenchmarkTools.jl package which among other things evaluates the function multiple times in order to reduce noise.

Tools

Julia and its package ecosystem includes tools that may help you diagnose problems and improve the performance of your code:

  • Profiling allows you to measure the performance of your running code and identify lines that serve as bottlenecks. For complex projects, the ProfileView package can help you visualize your profiling results.
  • The Traceur package can help you find common performance problems in your code.
  • Unexpectedly-large memory allocations–as reported by @time, @allocated, or the profiler (through calls to the garbage-collection routines)–hint that there might be issues with your code. If you don’t see another reason for the allocations, suspect a type problem. You can also start Julia with the --track-allocation=user option and examine the resulting *.mem files to see information about where those allocations occur. See Memory allocation analysis.
  • @code_warntype generates a representation of your code that can be helpful in finding expressions that result in type uncertainty. See @code_warntype below.

Avoid containers with abstract type parameters

When working with parameterized types, including arrays, it is best to avoid parameterizing with abstract types where possible.

Consider the following:

1. julia> a = Real[]
2. Real[]

4. julia> push!(a, 1); push!(a, 2.0); push!(a, π)
5. 3-element Vector{Real}:
6. 1
7. 2.0
8. π = 3.1415926535897...

Because a is an array of abstract type Real, it must be able to hold any Real value. Since Real objects can be of arbitrary size and structure, a must be represented as an array of pointers to individually allocated Real objects. However, if we instead only allow numbers of the same type, e.g. Float64, to be stored in a these can be stored more efficiently:

1. julia> a = Float64[]
2. Float64[]

4. julia> push!(a, 1); push!(a, 2.0); push!(a,  π)
5. 3-element Vector{Float64}:
6. 1.0
7. 2.0
8. 3.141592653589793

Assigning numbers into a will now convert them to Float64 and a will be stored as a contiguous block of 64-bit floating-point values that can be manipulated efficiently.

If you cannot avoid containers with abstract value types, it is sometimes better to parametrize with Any to avoid runtime type checking. E.g. IdDict{Any, Any} performs better than IdDict{Type, Vector}

See also the discussion under Parametric Types.

Type declarations

In many languages with optional type declarations, adding declarations is the principal way to make code run faster. This is not the case in Julia. In Julia, the compiler generally knows the types of all function arguments, local variables, and expressions. However, there are a few specific instances where declarations are helpful.

Avoid fields with abstract type

Types can be declared without specifying the types of their fields:

1. julia> struct MyAmbiguousType
2. a
3. end

This allows a to be of any type. This can often be useful, but it does have a downside: for objects of type MyAmbiguousType, the compiler will not be able to generate high-performance code. The reason is that the compiler uses the types of objects, not their values, to determine how to build code. Unfortunately, very little can be inferred about an object of type MyAmbiguousType:

1. julia> b = MyAmbiguousType("Hello")
2. MyAmbiguousType("Hello")

4. julia> c = MyAmbiguousType(17)
5. MyAmbiguousType(17)

7. julia> typeof(b)
8. MyAmbiguousType

10. julia> typeof(c)
11. MyAmbiguousType

The values of b and c have the same type, yet their underlying representation of data in memory is very different. Even if you stored just numeric values in field a, the fact that the memory representation of a UInt8 differs from a Float64 also means that the CPU needs to handle them using two different kinds of instructions. Since the required information is not available in the type, such decisions have to be made at run-time. This slows performance.

You can do better by declaring the type of a. Here, we are focused on the case where a might be any one of several types, in which case the natural solution is to use parameters. For example:

1. julia> mutable struct MyType{T<:AbstractFloat}
2. a::T
3. end

This is a better choice than

1. julia> mutable struct MyStillAmbiguousType
2. a::AbstractFloat
3. end

because the first version specifies the type of a from the type of the wrapper object. For example:

1. julia> m = MyType(3.2)
2. MyType{Float64}(3.2)

4. julia> t = MyStillAmbiguousType(3.2)
5. MyStillAmbiguousType(3.2)

7. julia> typeof(m)
8. MyType{Float64}

10. julia> typeof(t)
11. MyStillAmbiguousType

The type of field a can be readily determined from the type of m, but not from the type of t. Indeed, in t it’s possible to change the type of the field a:

1. julia> typeof(t.a)
2. Float64

4. julia> t.a = 4.5f0
5. 4.5f0

7. julia> typeof(t.a)
8. Float32

In contrast, once m is constructed, the type of m.a cannot change:

1. julia> m.a = 4.5f0
2. 4.5f0

4. julia> typeof(m.a)
5. Float64

The fact that the type of m.a is known from m‘s type—coupled with the fact that its type cannot change mid-function—allows the compiler to generate highly-optimized code for objects like m but not for objects like t.

Of course, all of this is true only if we construct m with a concrete type. We can break this by explicitly constructing it with an abstract type:

1. julia> m = MyType{AbstractFloat}(3.2)
2. MyType{AbstractFloat}(3.2)

4. julia> typeof(m.a)
5. Float64

7. julia> m.a = 4.5f0
8. 4.5f0

10. julia> typeof(m.a)
11. Float32

For all practical purposes, such objects behave identically to those of MyStillAmbiguousType.

It’s quite instructive to compare the sheer amount code generated for a simple function

1. func(m::MyType) = m.a+1

using

1. code_llvm(func, Tuple{MyType{Float64}})
2. code_llvm(func, Tuple{MyType{AbstractFloat}})

For reasons of length the results are not shown here, but you may wish to try this yourself. Because the type is fully-specified in the first case, the compiler doesn’t need to generate any code to resolve the type at run-time. This results in shorter and faster code.

Avoid fields with abstract containers

The same best practices also work for container types:

1. julia> struct MySimpleContainer{A<:AbstractVector}
2. a::A
3. end

5. julia> struct MyAmbiguousContainer{T}
6. a::AbstractVector{T}
7. end

For example:

1. julia> c = MySimpleContainer(1:3);

3. julia> typeof(c)
4. MySimpleContainer{UnitRange{Int64}}

6. julia> c = MySimpleContainer([1:3;]);

8. julia> typeof(c)
9. MySimpleContainer{Vector{Int64}}

11. julia> b = MyAmbiguousContainer(1:3);

13. julia> typeof(b)
14. MyAmbiguousContainer{Int64}

16. julia> b = MyAmbiguousContainer([1:3;]);

18. julia> typeof(b)
19. MyAmbiguousContainer{Int64}

For MySimpleContainer, the object is fully-specified by its type and parameters, so the compiler can generate optimized functions. In most instances, this will probably suffice.

While the compiler can now do its job perfectly well, there are cases where you might wish that your code could do different things depending on the element type of a. Usually the best way to achieve this is to wrap your specific operation (here, foo) in a separate function:

1. julia> function sumfoo(c::MySimpleContainer)
2. s = 0
3. for x in c.a
4. s += foo(x)
5. end
6. s
7. end
8. sumfoo (generic function with 1 method)

10. julia> foo(x::Integer) = x
11. foo (generic function with 1 method)

13. julia> foo(x::AbstractFloat) = round(x)
14. foo (generic function with 2 methods)

This keeps things simple, while allowing the compiler to generate optimized code in all cases.

However, there are cases where you may need to declare different versions of the outer function for different element types or types of the AbstractVector of the field a in MySimpleContainer. You could do it like this:

1. julia> function myfunc(c::MySimpleContainer{<:AbstractArray{<:Integer}})
2. return c.a[1]+1
3. end
4. myfunc (generic function with 1 method)

6. julia> function myfunc(c::MySimpleContainer{<:AbstractArray{<:AbstractFloat}})
7. return c.a[1]+2
8. end
9. myfunc (generic function with 2 methods)

11. julia> function myfunc(c::MySimpleContainer{Vector{T}}) where T <: Integer
12. return c.a[1]+3
13. end
14. myfunc (generic function with 3 methods)

1. julia> myfunc(MySimpleContainer(1:3))
2. 2

4. julia> myfunc(MySimpleContainer(1.0:3))
5. 3.0

7. julia> myfunc(MySimpleContainer([1:3;]))
8. 4

Annotate values taken from untyped locations

It is often convenient to work with data structures that may contain values of any type (arrays of type Array{Any}). But, if you’re using one of these structures and happen to know the type of an element, it helps to share this knowledge with the compiler:

1. function foo(a::Array{Any,1})
2. x = a[1]::Int32
3. b = x+1
4. ...
5. end

Here, we happened to know that the first element of a would be an Int32. Making an annotation like this has the added benefit that it will raise a run-time error if the value is not of the expected type, potentially catching certain bugs earlier.

In the case that the type of a[1] is not known precisely, x can be declared via x = convert(Int32, a[1])::Int32. The use of the convert function allows a[1] to be any object convertible to an Int32 (such as UInt8), thus increasing the genericity of the code by loosening the type requirement. Notice that convert itself needs a type annotation in this context in order to achieve type stability. This is because the compiler cannot deduce the type of the return value of a function, even convert, unless the types of all the function’s arguments are known.

Type annotation will not enhance (and can actually hinder) performance if the type is abstract, or constructed at run-time. This is because the compiler cannot use the annotation to specialize the subsequent code, and the type-check itself takes time. For example, in the code:

1. function nr(a, prec)
2. ctype = prec == 32 ? Float32 : Float64
3. b = Complex{ctype}(a)
4. c = (b + 1.0f0)::Complex{ctype}
5. abs(c)
6. end

the annotation of c harms performance. To write performant code involving types constructed at run-time, use the function-barrier technique discussed below, and ensure that the constructed type appears among the argument types of the kernel function so that the kernel operations are properly specialized by the compiler. For example, in the above snippet, as soon as b is constructed, it can be passed to another function k, the kernel. If, for example, function k declares b as an argument of type Complex{T}, where T is a type parameter, then a type annotation appearing in an assignment statement within k of the form:

1. c = (b + 1.0f0)::Complex{T}

does not hinder performance (but does not help either) since the compiler can determine the type of c at the time k is compiled.

Be aware of when Julia avoids specializing

As a heuristic, Julia avoids automatically specializing on argument type parameters in three specific cases: Type, Function, and Vararg. Julia will always specialize when the argument is used within the method, but not if the argument is just passed through to another function. This usually has no performance impact at runtime and improves compiler performance. If you find it does have a performance impact at runtime in your case, you can trigger specialization by adding a type parameter to the method declaration. Here are some examples:

This will not specialize:

1. function f_type(t)  # or t::Type
2. x = ones(t, 10)
3. return sum(map(sin, x))
4. end

but this will:

1. function g_type(t::Type{T}) where T
2. x = ones(T, 10)
3. return sum(map(sin, x))
4. end

These will not specialize:

1. f_func(f, num) = ntuple(f, div(num, 2))
2. g_func(g::Function, num) = ntuple(g, div(num, 2))

but this will:

1. h_func(h::H, num) where {H} = ntuple(h, div(num, 2))

This will not specialize:

1. f_vararg(x::Int...) = tuple(x...)

but this will:

1. g_vararg(x::Vararg{Int, N}) where {N} = tuple(x...)

One only needs to introduce a single type parameter to force specialization, even if the other types are unconstrained. For example, this will also specialize, and is useful when the arguments are not all of the same type:

1. h_vararg(x::Vararg{Any, N}) where {N} = tuple(x...)

Note that @code_typed and friends will always show you specialized code, even if Julia would not normally specialize that method call. You need to check the method internals if you want to see whether specializations are generated when argument types are changed, i.e., if (@which f(...)).specializations contains specializations for the argument in question.

Break functions into multiple definitions

Writing a function as many small definitions allows the compiler to directly call the most applicable code, or even inline it.

Here is an example of a “compound function” that should really be written as multiple definitions:

1. using LinearAlgebra

3. function mynorm(A)
4. if isa(A, Vector)
5. return sqrt(real(dot(A,A)))
6. elseif isa(A, Matrix)
7. return maximum(svdvals(A))
8. else
9. error("mynorm: invalid argument")
10. end
11. end

This can be written more concisely and efficiently as:

1. norm(x::Vector) = sqrt(real(dot(x, x)))
2. norm(A::Matrix) = maximum(svdvals(A))

It should however be noted that the compiler is quite efficient at optimizing away the dead branches in code written as the mynorm example.

Write “type-stable” functions

When possible, it helps to ensure that a function always returns a value of the same type. Consider the following definition:

1. pos(x) = x < 0 ? 0 : x

Although this seems innocent enough, the problem is that 0 is an integer (of type Int) and x might be of any type. Thus, depending on the value of x, this function might return a value of either of two types. This behavior is allowed, and may be desirable in some cases. But it can easily be fixed as follows:

1. pos(x) = x < 0 ? zero(x) : x

There is also a oneunit function, and a more general oftype(x, y) function, which returns y converted to the type of x.

Avoid changing the type of a variable

An analogous “type-stability” problem exists for variables used repeatedly within a function:

1. function foo()
2. x = 1
3. for i = 1:10
4. x /= rand()
5. end
6. return x
7. end

Local variable x starts as an integer, and after one loop iteration becomes a floating-point number (the result of / operator). This makes it more difficult for the compiler to optimize the body of the loop. There are several possible fixes:

  • Initialize x with x = 1.0
  • Declare the type of x explicitly as x::Float64 = 1
  • Use an explicit conversion by x = oneunit(Float64)
  • Initialize with the first loop iteration, to x = 1 / rand(), then loop for i = 2:10

Separate kernel functions (aka, function barriers)

Many functions follow a pattern of performing some set-up work, and then running many iterations to perform a core computation. Where possible, it is a good idea to put these core computations in separate functions. For example, the following contrived function returns an array of a randomly-chosen type:

1. julia> function strange_twos(n)
2. a = Vector{rand(Bool) ? Int64 : Float64}(undef, n)
3. for i = 1:n
4. a[i] = 2
5. end
6. return a
7. end;

9. julia> strange_twos(3)
10. 3-element Vector{Float64}:
11. 2.0
12. 2.0
13. 2.0

This should be written as:

1. julia> function fill_twos!(a)
2. for i = eachindex(a)
3. a[i] = 2
4. end
5. end;

7. julia> function strange_twos(n)
8. a = Vector{rand(Bool) ? Int64 : Float64}(undef, n)
9. fill_twos!(a)
10. return a
11. end;

13. julia> strange_twos(3)
14. 3-element Vector{Float64}:
15. 2.0
16. 2.0
17. 2.0

Julia’s compiler specializes code for argument types at function boundaries, so in the original implementation it does not know the type of a during the loop (since it is chosen randomly). Therefore the second version is generally faster since the inner loop can be recompiled as part of fill_twos! for different types of a.

The second form is also often better style and can lead to more code reuse.

This pattern is used in several places in Julia Base. For example, see vcat and hcat in abstractarray.jl, or the fill! function, which we could have used instead of writing our own fill_twos!.

Functions like strange_twos occur when dealing with data of uncertain type, for example data loaded from an input file that might contain either integers, floats, strings, or something else.

Types with values-as-parameters

Let’s say you want to create an N-dimensional array that has size 3 along each axis. Such arrays can be created like this:

1. julia> A = fill(5.0, (3, 3))
2. 3×3 Matrix{Float64}:
3. 5.0  5.0  5.0
4. 5.0  5.0  5.0
5. 5.0  5.0  5.0

This approach works very well: the compiler can figure out that A is an Array{Float64,2} because it knows the type of the fill value (5.0::Float64) and the dimensionality ((3, 3)::NTuple{2,Int}). This implies that the compiler can generate very efficient code for any future usage of A in the same function.

But now let’s say you want to write a function that creates a 3×3×… array in arbitrary dimensions; you might be tempted to write a function

1. julia> function array3(fillval, N)
2. fill(fillval, ntuple(d->3, N))
3. end
4. array3 (generic function with 1 method)

6. julia> array3(5.0, 2)
7. 3×3 Matrix{Float64}:
8. 5.0  5.0  5.0
9. 5.0  5.0  5.0
10. 5.0  5.0  5.0

This works, but (as you can verify for yourself using @code_warntype array3(5.0, 2)) the problem is that the output type cannot be inferred: the argument N is a value of type Int, and type-inference does not (and cannot) predict its value in advance. This means that code using the output of this function has to be conservative, checking the type on each access of A; such code will be very slow.

Now, one very good way to solve such problems is by using the function-barrier technique. However, in some cases you might want to eliminate the type-instability altogether. In such cases, one approach is to pass the dimensionality as a parameter, for example through Val{T}() (see “Value types”):

1. julia> function array3(fillval, ::Val{N}) where N
2. fill(fillval, ntuple(d->3, Val(N)))
3. end
4. array3 (generic function with 1 method)

6. julia> array3(5.0, Val(2))
7. 3×3 Matrix{Float64}:
8. 5.0  5.0  5.0
9. 5.0  5.0  5.0
10. 5.0  5.0  5.0

Julia has a specialized version of ntuple that accepts a Val{::Int} instance as the second parameter; by passing N as a type-parameter, you make its “value” known to the compiler. Consequently, this version of array3 allows the compiler to predict the return type.

However, making use of such techniques can be surprisingly subtle. For example, it would be of no help if you called array3 from a function like this:

1. function call_array3(fillval, n)
2. A = array3(fillval, Val(n))
3. end

Here, you’ve created the same problem all over again: the compiler can’t guess what n is, so it doesn’t know the type of Val(n). Attempting to use Val, but doing so incorrectly, can easily make performance worse in many situations. (Only in situations where you’re effectively combining Val with the function-barrier trick, to make the kernel function more efficient, should code like the above be used.)

An example of correct usage of Val would be:

1. function filter3(A::AbstractArray{T,N}) where {T,N}
2. kernel = array3(1, Val(N))
3. filter(A, kernel)
4. end

In this example, N is passed as a parameter, so its “value” is known to the compiler. Essentially, Val(T) works only when T is either hard-coded/literal (Val(3)) or already specified in the type-domain.

The dangers of abusing multiple dispatch (aka, more on types with values-as-parameters)

Once one learns to appreciate multiple dispatch, there’s an understandable tendency to go overboard and try to use it for everything. For example, you might imagine using it to store information, e.g.

1. struct Car{Make, Model}
2. year::Int
3. ...more fields...
4. end

and then dispatch on objects like Car{:Honda,:Accord}(year, args...).

This might be worthwhile when either of the following are true:

  • You require CPU-intensive processing on each Car, and it becomes vastly more efficient if you know the Make and Model at compile time and the total number of different Make or Model that will be used is not too large.
  • You have homogenous lists of the same type of Car to process, so that you can store them all in an Array{Car{:Honda,:Accord},N}.

When the latter holds, a function processing such a homogenous array can be productively specialized: Julia knows the type of each element in advance (all objects in the container have the same concrete type), so Julia can “look up” the correct method calls when the function is being compiled (obviating the need to check at run-time) and thereby emit efficient code for processing the whole list.

When these do not hold, then it’s likely that you’ll get no benefit; worse, the resulting “combinatorial explosion of types” will be counterproductive. If items[i+1] has a different type than item[i], Julia has to look up the type at run-time, search for the appropriate method in method tables, decide (via type intersection) which one matches, determine whether it has been JIT-compiled yet (and do so if not), and then make the call. In essence, you’re asking the full type- system and JIT-compilation machinery to basically execute the equivalent of a switch statement or dictionary lookup in your own code.

Some run-time benchmarks comparing (1) type dispatch, (2) dictionary lookup, and (3) a “switch” statement can be found on the mailing list.

Perhaps even worse than the run-time impact is the compile-time impact: Julia will compile specialized functions for each different Car{Make, Model}; if you have hundreds or thousands of such types, then every function that accepts such an object as a parameter (from a custom get_year function you might write yourself, to the generic push! function in Julia Base) will have hundreds or thousands of variants compiled for it. Each of these increases the size of the cache of compiled code, the length of internal lists of methods, etc. Excess enthusiasm for values-as-parameters can easily waste enormous resources.

Access arrays in memory order, along columns

Multidimensional arrays in Julia are stored in column-major order. This means that arrays are stacked one column at a time. This can be verified using the vec function or the syntax [:] as shown below (notice that the array is ordered [1 3 2 4], not [1 2 3 4]):

1. julia> x = [1 2; 3 4]
2. 2×2 Matrix{Int64}:
3. 1  2
4. 3  4

6. julia> x[:]
7. 4-element Vector{Int64}:
8. 1
9. 3
10. 2
11. 4

This convention for ordering arrays is common in many languages like Fortran, Matlab, and R (to name a few). The alternative to column-major ordering is row-major ordering, which is the convention adopted by C and Python (numpy) among other languages. Remembering the ordering of arrays can have significant performance effects when looping over arrays. A rule of thumb to keep in mind is that with column-major arrays, the first index changes most rapidly. Essentially this means that looping will be faster if the inner-most loop index is the first to appear in a slice expression. Keep in mind that indexing an array with : is an implicit loop that iteratively accesses all elements within a particular dimension; it can be faster to extract columns than rows, for example.

Consider the following contrived example. Imagine we wanted to write a function that accepts a Vector and returns a square Matrix with either the rows or the columns filled with copies of the input vector. Assume that it is not important whether rows or columns are filled with these copies (perhaps the rest of the code can be easily adapted accordingly). We could conceivably do this in at least four ways (in addition to the recommended call to the built-in repeat):

1. function copy_cols(x::Vector{T}) where T
2. inds = axes(x, 1)
3. out = similar(Array{T}, inds, inds)
4. for i = inds
5. out[:, i] = x
6. end
7. return out
8. end

10. function copy_rows(x::Vector{T}) where T
11. inds = axes(x, 1)
12. out = similar(Array{T}, inds, inds)
13. for i = inds
14. out[i, :] = x
15. end
16. return out
17. end

19. function copy_col_row(x::Vector{T}) where T
20. inds = axes(x, 1)
21. out = similar(Array{T}, inds, inds)
22. for col = inds, row = inds
23. out[row, col] = x[row]
24. end
25. return out
26. end

28. function copy_row_col(x::Vector{T}) where T
29. inds = axes(x, 1)
30. out = similar(Array{T}, inds, inds)
31. for row = inds, col = inds
32. out[row, col] = x[col]
33. end
34. return out
35. end

Now we will time each of these functions using the same random 10000 by 1 input vector:

1. julia> x = randn(10000);

3. julia> fmt(f) = println(rpad(string(f)*": ", 14, ' '), @elapsed f(x))

5. julia> map(fmt, [copy_cols, copy_rows, copy_col_row, copy_row_col]);
6. copy_cols:    0.331706323
7. copy_rows:    1.799009911
8. copy_col_row: 0.415630047
9. copy_row_col: 1.721531501

Notice that copy_cols is much faster than copy_rows. This is expected because copy_cols respects the column-based memory layout of the Matrix and fills it one column at a time. Additionally, copy_col_row is much faster than copy_row_col because it follows our rule of thumb that the first element to appear in a slice expression should be coupled with the inner-most loop.

Pre-allocating outputs

If your function returns an Array or some other complex type, it may have to allocate memory. Unfortunately, oftentimes allocation and its converse, garbage collection, are substantial bottlenecks.

Sometimes you can circumvent the need to allocate memory on each function call by preallocating the output. As a trivial example, compare

1. julia> function xinc(x)
2. return [x, x+1, x+2]
3. end;

5. julia> function loopinc()
6. y = 0
7. for i = 1:10^7
8. ret = xinc(i)
9. y += ret[2]
10. end
11. return y
12. end;

with

1. julia> function xinc!(ret::AbstractVector{T}, x::T) where T
2. ret[1] = x
3. ret[2] = x+1
4. ret[3] = x+2
5. nothing
6. end;

8. julia> function loopinc_prealloc()
9. ret = Vector{Int}(undef, 3)
10. y = 0
11. for i = 1:10^7
12. xinc!(ret, i)
13. y += ret[2]
14. end
15. return y
16. end;

Timing results:

1. julia> @time loopinc()
2. 0.529894 seconds (40.00 M allocations: 1.490 GiB, 12.14% gc time)
3. 50000015000000

5. julia> @time loopinc_prealloc()
6. 0.030850 seconds (6 allocations: 288 bytes)
7. 50000015000000

Preallocation has other advantages, for example by allowing the caller to control the “output” type from an algorithm. In the example above, we could have passed a SubArray rather than an Array, had we so desired.

Taken to its extreme, pre-allocation can make your code uglier, so performance measurements and some judgment may be required. However, for “vectorized” (element-wise) functions, the convenient syntax x .= f.(y) can be used for in-place operations with fused loops and no temporary arrays (see the dot syntax for vectorizing functions).

More dots: Fuse vectorized operations

Julia has a special dot syntax that converts any scalar function into a “vectorized” function call, and any operator into a “vectorized” operator, with the special property that nested “dot calls” are fusing: they are combined at the syntax level into a single loop, without allocating temporary arrays. If you use .= and similar assignment operators, the result can also be stored in-place in a pre-allocated array (see above).

In a linear-algebra context, this means that even though operations like vector + vector and vector * scalar are defined, it can be advantageous to instead use vector .+ vector and vector .* scalar because the resulting loops can be fused with surrounding computations. For example, consider the two functions:

1. julia> f(x) = 3x.^2 + 4x + 7x.^3;

3. julia> fdot(x) = @. 3x^2 + 4x + 7x^3 # equivalent to 3 .* x.^2 .+ 4 .* x .+ 7 .* x.^3;

Both f and fdot compute the same thing. However, fdot (defined with the help of the @. macro) is significantly faster when applied to an array:

1. julia> x = rand(10^6);

3. julia> @time f(x);
4. 0.019049 seconds (16 allocations: 45.777 MiB, 18.59% gc time)

6. julia> @time fdot(x);
7. 0.002790 seconds (6 allocations: 7.630 MiB)

9. julia> @time f.(x);
10. 0.002626 seconds (8 allocations: 7.630 MiB)

That is, fdot(x) is ten times faster and allocates 1/6 the memory of f(x), because each * and + operation in f(x) allocates a new temporary array and executes in a separate loop. (Of course, if you just do f.(x) then it is as fast as fdot(x) in this example, but in many contexts it is more convenient to just sprinkle some dots in your expressions rather than defining a separate function for each vectorized operation.)

Consider using views for slices

In Julia, an array “slice” expression like array[1:5, :] creates a copy of that data (except on the left-hand side of an assignment, where array[1:5, :] = ... assigns in-place to that portion of array). If you are doing many operations on the slice, this can be good for performance because it is more efficient to work with a smaller contiguous copy than it would be to index into the original array. On the other hand, if you are just doing a few simple operations on the slice, the cost of the allocation and copy operations can be substantial.

An alternative is to create a “view” of the array, which is an array object (a SubArray) that actually references the data of the original array in-place, without making a copy. (If you write to a view, it modifies the original array’s data as well.) This can be done for individual slices by calling view, or more simply for a whole expression or block of code by putting @views in front of that expression. For example:

1. julia> fcopy(x) = sum(x[2:end-1]);

3. julia> @views fview(x) = sum(x[2:end-1]);

5. julia> x = rand(10^6);

7. julia> @time fcopy(x);
8. 0.003051 seconds (3 allocations: 7.629 MB)

10. julia> @time fview(x);
11. 0.001020 seconds (1 allocation: 16 bytes)

Notice both the 3× speedup and the decreased memory allocation of the fview version of the function.

Copying data is not always bad

Arrays are stored contiguously in memory, lending themselves to CPU vectorization and fewer memory accesses due to caching. These are the same reasons that it is recommended to access arrays in column-major order (see above). Irregular access patterns and non-contiguous views can drastically slow down computations on arrays because of non-sequential memory access.

Copying irregularly-accessed data into a contiguous array before operating on it can result in a large speedup, such as in the example below. Here, a matrix and a vector are being accessed at 800,000 of their randomly-shuffled indices before being multiplied. Copying the views into plain arrays speeds up the multiplication even with the cost of the copying operation.

1. julia> using Random

3. julia> x = randn(1_000_000);

5. julia> inds = shuffle(1:1_000_000)[1:800000];

7. julia> A = randn(50, 1_000_000);

9. julia> xtmp = zeros(800_000);

11. julia> Atmp = zeros(50, 800_000);

13. julia> @time sum(view(A, :, inds) * view(x, inds))
14. 0.412156 seconds (14 allocations: 960 bytes)
15. -4256.759568345458

17. julia> @time begin
18. copyto!(xtmp, view(x, inds))
19. copyto!(Atmp, view(A, :, inds))
20. sum(Atmp * xtmp)
21. end
22. 0.285923 seconds (14 allocations: 960 bytes)
23. -4256.759568345134

Provided there is enough memory for the copies, the cost of copying the view to an array is far outweighed by the speed boost from doing the matrix multiplication on a contiguous array.

Consider StaticArrays.jl for small fixed-size vector/matrix operations

If your application involves many small (< 100 element) arrays of fixed sizes (i.e. the size is known prior to execution), then you might want to consider using the StaticArrays.jl package. This package allows you to represent such arrays in a way that avoids unnecessary heap allocations and allows the compiler to specialize code for the size of the array, e.g. by completely unrolling vector operations (eliminating the loops) and storing elements in CPU registers.

For example, if you are doing computations with 2d geometries, you might have many computations with 2-component vectors. By using the SVector type from StaticArrays.jl, you can use convenient vector notation and operations like norm(3v - w) on vectors v and w, while allowing the compiler to unroll the code to a minimal computation equivalent to @inbounds hypot(3v[1]-w[1], 3v[2]-w[2]).

Avoid string interpolation for I/O

When writing data to a file (or other I/O device), forming extra intermediate strings is a source of overhead. Instead of:

1. println(file, "$a $b")

use:

1. println(file, a, " ", b)

The first version of the code forms a string, then writes it to the file, while the second version writes values directly to the file. Also notice that in some cases string interpolation can be harder to read. Consider:

1. println(file, "$(f(a))$(f(b))")

versus:

1. println(file, f(a), f(b))

Optimize network I/O during parallel execution

When executing a remote function in parallel:

1. using Distributed

3. responses = Vector{Any}(undef, nworkers())
4. @sync begin
5. for (idx, pid) in enumerate(workers())
6. @async responses[idx] = remotecall_fetch(foo, pid, args...)
7. end
8. end

is faster than:

1. using Distributed

3. refs = Vector{Any}(undef, nworkers())
4. for (idx, pid) in enumerate(workers())
5. refs[idx] = @spawnat pid foo(args...)
6. end
7. responses = [fetch(r) for r in refs]

The former results in a single network round-trip to every worker, while the latter results in two network calls - first by the @spawnat and the second due to the fetch (or even a wait). The fetch/wait is also being executed serially resulting in an overall poorer performance.

Fix deprecation warnings

A deprecated function internally performs a lookup in order to print a relevant warning only once. This extra lookup can cause a significant slowdown, so all uses of deprecated functions should be modified as suggested by the warnings.

Tweaks

These are some minor points that might help in tight inner loops.

Performance Annotations

Sometimes you can enable better optimization by promising certain program properties.

  • Use @inbounds to eliminate array bounds checking within expressions. Be certain before doing this. If the subscripts are ever out of bounds, you may suffer crashes or silent corruption.
  • Use @fastmath to allow floating point optimizations that are correct for real numbers, but lead to differences for IEEE numbers. Be careful when doing this, as this may change numerical results. This corresponds to the -ffast-math option of clang.
  • Write [@simd](/read/julialang-1.6.0-en/b865a5c21fdb4db6.md#Base.

（本文超出站点单文档体积上限，已截断）