gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[PYTHON] Removed -std=gnu++11 in extra_cflags

Browse Source
This commit is contained in:
Philippe Tillet
2020-05-09 22:57:40 -04:00
committed by Philippe Tillet
parent 04a9ea060b
commit 1426b103e9
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
python/triton/kernel.py
View File

@@ -85,13 +85,13 @@ def _make_framework_op(arg_types):
ccdir = os.path.join(libtriton.__file__, os.path.pardir) ccdir = os.path.join(libtriton.__file__, os.path.pardir)
ccdir = os.path.realpath(ccdir) ccdir = os.path.realpath(ccdir)
print('[TRITON] Compiling op...') print('[TRITON] Compiling op...')
lib = torch.utils.cpp_extension.load_inline(name, src, lib = torch.utils.cpp_extension.load(name, cpp,
extra_ldflags = [f'-L{ccdir}', '-ltriton'], extra_ldflags = [f'-L{ccdir}', '-ltriton'],
extra_include_paths = [os.path.join(ccdir, 'include')], extra_include_paths = [os.path.join(ccdir, 'include')],
extra_cflags = [f'-std=gnu++11'], #extra_cflags = [f'-std=gnu++11'],
build_directory = root, build_directory = root,
is_python_module = False, is_python_module = False,
with_cuda = True) with_cuda = True)
fw.torch.ops.load_library(so) fw.torch.ops.load_library(so)
return getattr(fw.torch.ops.triton, name) return getattr(fw.torch.ops.triton, name)