cory-master/BabelStream
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
418315543c
BabelStream/JuliaStream.jl
History
Tom Lin d675875dcd Switch back to -p for DistributedStream
2021-06-30 19:03:39 +01:00
..
src Switch back to -p for DistributedStream 2021-06-30 19:03:39 +01:00
.gitignore Initial Julia implementation 2021-06-10 04:20:40 +01:00
.JuliaFormatter.toml Initial Julia implementation 2021-06-10 04:20:40 +01:00
Manifest.toml Initial Julia implementation 2021-06-10 04:20:40 +01:00
Project.toml Initial Julia implementation 2021-06-10 04:20:40 +01:00
README.md Switch back to -p for DistributedStream 2021-06-30 19:03:39 +01:00

README.md

JuliaStream.jl

This is an implementation of BabelStream in Julia which contains the following variants:

  • PlainStream.jl - Single threaded for
  • ThreadedStream.jl - Threaded implementation with Threads.@threads macros
  • DistributedStream.jl - Process based parallelism with @distributed macros
  • CUDAStream.jl - Direct port of BabelStream's native CUDA implementation using CUDA.jl
  • AMDGPUStream.jl - Direct port of BabelStream's native HIP implementation using AMDGPU.jl

Build & Run

Prerequisites

  • Julia >= 1.6+

With Julia on path, run the benchmark with:

> cd JuliaStream.jl
> julia --project -e 'import Pkg; Pkg.instantiate()' # only required on first run
> julia --project src/<IMPL>Stream.jl

Important:

  • Julia is 1-indexed, so N >= 1 in --device N.
  • Thread count for ThreadedStream must be set via the JULIA_NUM_THREADS environment variable (e.g export JULIA_NUM_THREADS=$(nproc)) otherwise it defaults to 1.
  • Worker count for DistributedStream is set with -p <N> as per the documentation.
  • Certain implementations such as CUDA and AMDGPU will do hardware detection at runtime and may download and/or compile further software packages for the platform.