Skip to content

Commit 807c8e8

Browse files
committed
Fix meson build default logic and update sm90 in README
Signed-off-by: Michal Shalev <[email protected]>
1 parent cf42725 commit 807c8e8

File tree

2 files changed

+10
-3
lines changed

2 files changed

+10
-3
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,7 @@ Common CUDA compute capabilities:
153153
- `compute_80,code=sm_80`: NVIDIA Ampere (A100, RTX 30xx)
154154
- `compute_86,code=sm_86`: NVIDIA Ampere (RTX 30xx consumer)
155155
- `compute_89,code=sm_89`: NVIDIA Ada Lovelace (RTX 40xx)
156-
- `compute_90,code=sm_90`: NVIDIA Hopper (H100)
156+
- `compute_90,code=sm_90`: NVIDIA Hopper (H100, H800, H200)
157157

158158
### Building Documentation
159159

meson.build

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -92,22 +92,29 @@ if cuda_dep.found()
9292
cuda_args_option = get_option('cuda_args')
9393
cuda_link_args_option = get_option('cuda_link_args')
9494

95-
cuda_args = [
95+
# Define default architectures
96+
default_cuda_args = [
9697
'-gencode=arch=compute_80,code=sm_80',
9798
'-gencode=arch=compute_90,code=sm_90'
9899
]
99-
cuda_link_args = [
100+
default_cuda_link_args = [
100101
'-gencode=arch=compute_80,code=sm_80',
101102
'-gencode=arch=compute_90,code=sm_90'
102103
]
103104

104105
# Apply defaults only if user hasn't specified custom args
105106
if cuda_args_option == []
107+
cuda_args = default_cuda_args
106108
add_project_arguments(cuda_args, language: 'cuda')
109+
else
110+
cuda_args = []
107111
endif
108112

109113
if cuda_link_args_option == []
114+
cuda_link_args = default_cuda_link_args
110115
add_project_link_arguments(cuda_link_args, language: 'cuda')
116+
else
117+
cuda_link_args = []
111118
endif
112119

113120
# Refer to https://mesonbuild.com/Cuda-module.html

0 commit comments

Comments
 (0)