Skip to content

Commit

Permalink
Add mxp versions of nvidia-hpl workloads
Browse files Browse the repository at this point in the history
  • Loading branch information
douglasjacobsen committed Dec 17, 2024
1 parent 5cbe333 commit 37bc23f
Showing 1 changed file with 35 additions and 3 deletions.
38 changes: 35 additions & 3 deletions var/ramble/repos/builtin/applications/nvidia-hpl/application.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,12 +38,29 @@ class NvidiaHpl(HplBase):
"execute", "./hpl.sh --dat {experiment_run_dir}/HPL.dat", use_mpi=True
)

executable(
"execute-mxp",
'./hpl-mxp.sh --gpu-affinity "{gpu_affinity}" --n {Ns} --nb {block_size} --nprow {Ps} --npcol {Qs} --nporder {nporder}',
use_mpi=True,
)

workload("standard", executables=["execute"])
workload("calculator", executables=["execute"])

workload_group("standard", workloads=["standard"], mode="append")
workload_group("calculator", workloads=["calculator"], mode="append")
workload_group("all_workloads", workloads=["standard", "calculator"])
workload("standard-mxp", executables=["execute-mxp"])
workload("calculator-mxp", executables=["execute-mxp"])

workload_group(
"standard", workloads=["standard", "standard-mxp"], mode="append"
)
workload_group(
"calculator", workloads=["calculator", "calculator-mxp"], mode="append"
)
workload_group(
"all_workloads",
workloads=["standard", "standard-mxp", "calculator", "calculator-mxp"],
)
workload_group("mxp", workloads=["standard-mxp", "calculator-mxp"])

workload_variable(
"nvshmem_disable_cuda_vmm",
Expand Down Expand Up @@ -168,3 +185,18 @@ class NvidiaHpl(HplBase):
description="Size of each block",
workload_group="calculator",
)

workload_variable(
"nporder",
default="row",
description="Major order to use for matrix",
values=["row", "column"],
workload_group="mxp",
)

workload_variable(
"gpu_affinity",
default="0:1:2:3:4:5:6:7",
description="Colon delimited list of GPU IDs",
workload_group="mxp",
)

0 comments on commit 37bc23f

Please sign in to comment.