Skip to content

Commit 6e6e5fb

Browse files
committed
Added specification to CUPY compiler options
1 parent a020791 commit 6e6e5fb

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

kernel_tuner/backends/cupy.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -125,6 +125,10 @@ def compile(self, kernel_instance):
125125
compiler_options = self.compiler_options
126126
if not any(["-std=" in opt for opt in self.compiler_options]):
127127
compiler_options = ["--std=c++11"] + self.compiler_options
128+
if not any([b"--gpu-architecture=" in opt for opt in compiler_options]):
129+
compiler_options.append(
130+
f"--gpu-architecture=compute_{self.cc}".encode("UTF-8")
131+
)
128132

129133
options = tuple(compiler_options)
130134

0 commit comments

Comments
 (0)