change extra_compile_flag to -std=c++14: setup.py

pytorch_binding/setup.py

@@ -1,11 +1,17 @@
+from distutils.version import LooseVersion
 import os
 import platform
 import sys
 from setuptools import setup, find_packages
 import torch
 from torch.utils.cpp_extension import BuildExtension, CppExtension
 
-extra_compile_args = ['-std=c++11', '-fPIC']
+
+extra_compile_args = ['-fPIC']
+if LooseVersion(torch.__version__) >= LooseVersion("1.5.0"):
+    extra_compile_args += ['-std=c++14']
+else:
+    extra_compile_args += ['-std=c++11']
 warp_rnnt_path = "../build"
 
 if torch.cuda.is_available() or "CUDA_HOME" in os.environ: