diff --git a/setup.py b/setup.py index e842302548..bf2ab7a406 100644 --- a/setup.py +++ b/setup.py @@ -523,12 +523,12 @@ def get_extensions(): "-DNDEBUG" if not debug_mode else "-DDEBUG", "-O3" if not debug_mode else "-O0", "-t=0", - "-std=c++17", + "-std=c++20", ] rocm_args = [ "-DNDEBUG" if not debug_mode else "-DDEBUG", "-O3" if not debug_mode else "-O0", - "-std=c++17", + "-std=c++20", ] maybe_hipify_v2_flag = [] if use_rocm and detect_hipify_v2(): @@ -772,7 +772,7 @@ def get_extensions(): extra_compile_args={ "cxx": [ f"-DPy_LIMITED_API={min_supported_cpython_hexcode}", - "-std=c++17", + "-std=c++20", "-O3", ], "nvcc": nvcc_args diff --git a/torchao/csrc/cpu/CMakeLists.txt b/torchao/csrc/cpu/CMakeLists.txt index eee77d77ac..03a5dc4751 100644 --- a/torchao/csrc/cpu/CMakeLists.txt +++ b/torchao/csrc/cpu/CMakeLists.txt @@ -9,7 +9,7 @@ include(CMakeDependentOption) project(torchao) -set(CMAKE_CXX_STANDARD 17) +set(CMAKE_CXX_STANDARD 20) if (NOT CMAKE_BUILD_TYPE) set(CMAKE_BUILD_TYPE Release) diff --git a/torchao/experimental/CMakeLists.txt b/torchao/experimental/CMakeLists.txt index 84582f704e..70bd6d9063 100644 --- a/torchao/experimental/CMakeLists.txt +++ b/torchao/experimental/CMakeLists.txt @@ -9,7 +9,7 @@ include(CMakeDependentOption) project(torchao) -set(CMAKE_CXX_STANDARD 17) +set(CMAKE_CXX_STANDARD 20) if (NOT CMAKE_BUILD_TYPE) set(CMAKE_BUILD_TYPE Release) diff --git a/torchao/experimental/kernels/mps/test/Makefile b/torchao/experimental/kernels/mps/test/Makefile index 3c0da54f7c..2a95a98b21 100644 --- a/torchao/experimental/kernels/mps/test/Makefile +++ b/torchao/experimental/kernels/mps/test/Makefile @@ -1,7 +1,7 @@ all: test_lowbit test_lowbit: test_lowbit.mm ../src/OperationUtils.mm - clang++ -I${TORCHAO_ROOT} -O3 -std=c++17 -Wall -Wextra -o $@ $^ -framework Metal -framework Foundation + clang++ -I${TORCHAO_ROOT} -O3 -std=c++20 -Wall -Wextra -o $@ $^ -framework Metal -framework Foundation run: test_lowbit ./test_lowbit diff --git a/torchao/experimental/ops/mps/CMakeLists.txt b/torchao/experimental/ops/mps/CMakeLists.txt index 34d4b178e0..b6ef35759a 100644 --- a/torchao/experimental/ops/mps/CMakeLists.txt +++ b/torchao/experimental/ops/mps/CMakeLists.txt @@ -8,7 +8,7 @@ cmake_minimum_required(VERSION 3.19) project(torchao_ops_mps_linear_fp_act_xbit_weight) -set(CMAKE_CXX_STANDARD 17) +set(CMAKE_CXX_STANDARD 20) set(CMAKE_CXX_STANDARD_REQUIRED YES) if (NOT CMAKE_BUILD_TYPE)