File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -22,8 +22,10 @@ Call with `device_info(CUDA.name(CUDA.device()))`
22
22
"""
23
23
function device_info (device_name)
24
24
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
27
29
)
28
30
is_cuda = ClimaComms. device () isa ClimaComms. CUDADevice
29
31
if is_cuda && haskey (device_specs, device_name)
You can’t perform that action at this time.
0 commit comments