From d3a0c870d81af5d9af2435b6c253b17f17aa99b8 Mon Sep 17 00:00:00 2001
From: Peter Baylies <pbaylies@gmail.com>
Date: Mon, 21 Dec 2020 02:53:52 -0500
Subject: [PATCH] Update flags for nvcc.
---
dnnlib/tflib/custom_ops.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/dnnlib/tflib/custom_ops.py b/dnnlib/tflib/custom_ops.py
index ed31b76..b97691c 100755
--- a/dnnlib/tflib/custom_ops.py
+++ b/dnnlib/tflib/custom_ops.py
@@ -69,7 +69,7 @@ def _run_cmd(cmd):
raise RuntimeError('NVCC returned an error. See below for full command line and output log:\n\n%s\n\n%s' % (cmd, output))
def _prepare_nvcc_cli(opts):
- cmd = 'nvcc ' + opts.strip()
+ cmd = 'nvcc --std=c++11 -DNDEBUG ' + opts.strip()
cmd += ' --disable-warnings'
cmd += ' --include-path "%s"' % tf.sysconfig.get_include()
cmd += ' --include-path "%s"' % os.path.join(tf.sysconfig.get_include(), 'external', 'protobuf_archive', 'src')
--
GitLab