From b7db15ce944253ea587cc537fda3553fcbcc5df7 Mon Sep 17 00:00:00 2001 From: YLGH Date: Tue, 25 Mar 2025 14:15:39 -0700 Subject: [PATCH] Update nvcc flag c++20 Needed for fconcepts --- deep_gemm/jit/compiler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deep_gemm/jit/compiler.py b/deep_gemm/jit/compiler.py index aad8939..fec2eb9 100644 --- a/deep_gemm/jit/compiler.py +++ b/deep_gemm/jit/compiler.py @@ -96,7 +96,7 @@ def put(path, data, is_binary=False): def build(name: str, arg_defs: tuple, code: str) -> Runtime: # Compiler flags - nvcc_flags = ['-std=c++17', '-shared', '-O3', '--expt-relaxed-constexpr', '--expt-extended-lambda', + nvcc_flags = ['-std=c++20', '-shared', '-O3', '--expt-relaxed-constexpr', '--expt-extended-lambda', '-gencode=arch=compute_90a,code=sm_90a', '--ptxas-options=--register-usage-level=10' + (',--verbose' if 'DG_PTXAS_VERBOSE' in os.environ else ''), # Suppress some unnecessary warnings, such as unused variables for certain `constexpr` branch cases