Skip to content

Commit 4788064

Browse files
Merge pull request #1973 from CliMA/ck/cuda_device
Add cuda devices and references
2 parents add3b1b + f8a3a28 commit 4788064

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

benchmarks/scripts/benchmark_utils.jl

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,10 @@ Call with `device_info(CUDA.name(CUDA.device()))`
2222
"""
2323
function device_info(device_name)
2424
device_specs = Dict(
25-
"NVIDIA A100-SXM4-80GB" => (; device_bandwidth_GBs = 2_039),
26-
"Tesla P100-PCIE-16GB" => (; device_bandwidth_GBs = 732),
25+
"NVIDIA A100-SXM4-80GB" => (; device_bandwidth_GBs = 2_039), # https://www.nvidia.com/en-us/data-center/a100/
26+
"Tesla P100-PCIE-16GB" => (; device_bandwidth_GBs = 732), # https://images.nvidia.com/content/tesla/pdf/nvidia-tesla-p100-PCIe-datasheet.pdf
27+
"NVIDIA H100 80GB HBM3" => (; device_bandwidth_GBs = 3_350), # https://www.nvidia.com/en-us/data-center/h100/
28+
"NVIDIA GeForce GTX 1050" => (; device_bandwidth_GBs = 112.1), # https://www.techpowerup.com/gpu-specs/geforce-gtx-1050.c2875
2729
)
2830
is_cuda = ClimaComms.device() isa ClimaComms.CUDADevice
2931
if is_cuda && haskey(device_specs, device_name)

0 commit comments

Comments
 (0)