add -cubin

cupy · Oct 2, 2019 · 8415293 · 8415293
1 parent d4ff126
commit 8415293
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cupy/cuda/compiler.py b/cupy/cuda/compiler.py
@@ -125,7 +125,7 @@ def compile_using_nvcc(source, options=(), arch=None,
         raise ValueError('Invalid code_type %s. Should be cubin or ptx')
 
     arch_str = '-gencode=arch=compute_{cc},code=sm_{cc}'.format(cc=arch)
-    cmd = ['nvcc', '--%s' % code_type, arch_str] + list(options)
+    cmd = ['nvcc', '-cubin', '--%s' % code_type, arch_str] + list(options)
 
     with TemporaryDirectory() as root_dir:
         first_part = filename.split('.')[0]