updated c++17 compiler flags (#4983)

2026-01-29 21:49:54 +00:00 · 2023-10-27 18:19:56 +08:00
parent 1db6727678
commit 4e4a10c97d
5 changed files with 15 additions and 3 deletions
--- a/op_builder/multi_head_attn.py
+++ b/op_builder/multi_head_attn.py
@@ -35,6 +35,7 @@ class MultiHeadAttnBuilder(Builder):
    def nvcc_flags(self):
        extra_cuda_flags = [
            "-std=c++14",
+            "-std=c++17",
            "-U__CUDA_NO_HALF_OPERATORS__",
            "-U__CUDA_NO_HALF_CONVERSIONS__",
            "-U__CUDA_NO_HALF2_OPERATORS__",