From 1426b103e919f5dc3e35aed2b6e769baceee66f4 Mon Sep 17 00:00:00 2001 From: Philippe Tillet Date: Sat, 9 May 2020 22:57:40 -0400 Subject: [PATCH] [PYTHON] Removed -std=gnu++11 in extra_cflags --- python/triton/kernel.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/python/triton/kernel.py b/python/triton/kernel.py index 752bb30a7..6ef630c2a 100644 --- a/python/triton/kernel.py +++ b/python/triton/kernel.py @@ -85,13 +85,13 @@ def _make_framework_op(arg_types): ccdir = os.path.join(libtriton.__file__, os.path.pardir) ccdir = os.path.realpath(ccdir) print('[TRITON] Compiling op...') - lib = torch.utils.cpp_extension.load_inline(name, src, - extra_ldflags = [f'-L{ccdir}', '-ltriton'], - extra_include_paths = [os.path.join(ccdir, 'include')], - extra_cflags = [f'-std=gnu++11'], - build_directory = root, - is_python_module = False, - with_cuda = True) + lib = torch.utils.cpp_extension.load(name, cpp, + extra_ldflags = [f'-L{ccdir}', '-ltriton'], + extra_include_paths = [os.path.join(ccdir, 'include')], + #extra_cflags = [f'-std=gnu++11'], + build_directory = root, + is_python_module = False, + with_cuda = True) fw.torch.ops.load_library(so) return getattr(fw.torch.ops.triton, name)