Skip to content

Commit

Permalink
Test
Browse files Browse the repository at this point in the history
  • Loading branch information
Sbozzolo committed Feb 14, 2024
1 parent a91793c commit ceabf07
Show file tree
Hide file tree
Showing 2 changed files with 43 additions and 40 deletions.
10 changes: 5 additions & 5 deletions .buildkite/JuliaProject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -3,12 +3,12 @@ CUDA_Runtime_jll = "76a88914-d11a-5bdc-97e0-2f5a05c973a2"
CUDA_Driver_jll = "4ee394cb-3365-5eb0-8335-949819d2adfc"
MPIPreferences = "3da0fdf6-3ccc-4f1b-acd9-58baa6c99267"

[preferences.CUDA_Driver_jll]
compat = false
# [preferences.CUDA_Driver_jll]
# compat = false

[preferences.CUDA_Runtime_jll]
version = "12.2"
local = "true"
# [preferences.CUDA_Runtime_jll]
# version = "12.2"
# local = "true"

[preferences.MPIPreferences]
_format = "1.0"
Expand Down
73 changes: 38 additions & 35 deletions .buildkite/pipeline.yml
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
agents:
queue: central
slurm_mem: 8G
modules: julia/1.10.0 cuda/12.2 ucx/1.14.1_cuda-12.2 openmpi/4.1.5_cuda-12.2 nsight-systems/2023.3.1
modules: julia/1.10.0 # cuda/12.2 ucx/1.14.1_cuda-12.2 openmpi/4.1.5_cuda-12.2 nsight-systems/2023.3.1

env:
JULIA_LOAD_PATH: "${JULIA_LOAD_PATH}:${BUILDKITE_BUILD_CHECKOUT_PATH}/.buildkite"
Expand All @@ -28,6 +28,7 @@ steps:
- echo "--- Instantiate perf environment"
- "julia --project=perf -e 'using Pkg; Pkg.instantiate(;verbose=true); Pkg.precompile(;strict=true)'"
- "julia --project=perf -e 'using Pkg; Pkg.status()'"
- "julia --project=perf -e 'using CUDA; println(CUDA.versioninfo())'"

agents:
slurm_cpus_per_task: 8
Expand All @@ -41,52 +42,54 @@ steps:
- group: "Unit tests"
steps:

- label: "CPU unit tests"
command: "julia --color=yes --project=perf test/runtests.jl"
# - label: "CPU unit tests"
# command: "julia --color=yes --project=perf test/runtests.jl"

- label: "Threaded CPU unit tests"
command: "julia --threads 8 --color=yes --project=perf test/runtests.jl"
agents:
slurm_ntasks: 8
slurm_mem: 48G
# - label: "Threaded CPU unit tests"
# command: "julia --threads 8 --color=yes --project=perf test/runtests.jl"
# agents:
# slurm_ntasks: 8
# slurm_mem: 48G

- label: "GPU unit tests"
command: "julia --color=yes --project=perf test/runtests.jl"
command:
- "julia --project=perf -e 'using CUDA; println(CUDA.versioninfo())'"
- "julia --color=yes --project=perf test/runtests.jl"
agents:
slurm_gpus: 1

- group: "Flame graphs"
steps:
# - group: "Flame graphs"
# steps:

- label: "Flame Graph: Gray Atm"
command: "julia --color=yes --project=perf perf/flame.jl gray_atm.jl"
artifact_paths: "flame_graphs/gray_atm/*"
# - label: "Flame Graph: Gray Atm"
# command: "julia --color=yes --project=perf perf/flame.jl gray_atm.jl"
# artifact_paths: "flame_graphs/gray_atm/*"

- label: "Flame Graph: Clear Sky"
command: "julia --color=yes --project=perf perf/flame.jl clear_sky.jl"
artifact_paths: "flame_graphs/clear_sky/*"
# - label: "Flame Graph: Clear Sky"
# command: "julia --color=yes --project=perf perf/flame.jl clear_sky.jl"
# artifact_paths: "flame_graphs/clear_sky/*"


- label: "Flame Graph: All Sky"
command: "julia --color=yes --project=perf perf/flame.jl all_sky.jl"
artifact_paths: "flame_graphs/all_sky/*"
# - label: "Flame Graph: All Sky"
# command: "julia --color=yes --project=perf perf/flame.jl all_sky.jl"
# artifact_paths: "flame_graphs/all_sky/*"

- group: "Benchmarks"
steps:
# - group: "Benchmarks"
# steps:

- label: "Benchmarks"
command: "julia --color=yes --project=perf perf/benchmark.jl"
# - label: "Benchmarks"
# command: "julia --color=yes --project=perf perf/benchmark.jl"

- label: "Benchmarks (Float32)"
command: "julia --color=yes --project=perf perf/benchmark.jl Float32"
# - label: "Benchmarks (Float32)"
# command: "julia --color=yes --project=perf perf/benchmark.jl Float32"

- label: "Threaded benchmarks"
command: "julia --threads=8 --color=yes --project=perf perf/benchmark.jl"
agents:
slurm_ntasks: 8
slurm_mem: 48G
# - label: "Threaded benchmarks"
# command: "julia --threads=8 --color=yes --project=perf perf/benchmark.jl"
# agents:
# slurm_ntasks: 8
# slurm_mem: 48G

- label: "GPU benchmarks"
command: "julia --color=yes --project=perf perf/benchmark.jl"
agents:
slurm_gpus: 1
# - label: "GPU benchmarks"
# command: "julia --color=yes --project=perf perf/benchmark.jl"
# agents:
# slurm_gpus: 1

0 comments on commit ceabf07

Please sign in to comment.