Skip to content

Commit 3097c66

Browse files
authored
Apply suggestions from code review
1 parent 9bdae0a commit 3097c66

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

cuda_core/examples/jit_lto_fractal.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ def link(self, user_code, target_type):
8888
program_options = self.program_options
8989
linker_options = LinkerOptions()
9090
else:
91-
raise AssertionError
91+
raise AssertionError(f"Invalid {target_type=}")
9292

9393
# First, user-defined code is compiled into a PTX object code
9494
user_object_code = Program(user_code, "c++", options=program_options).compile(target_type)
@@ -106,7 +106,7 @@ def run(self, kernel):
106106
launch(kernel, self.config, self.buffer.data.ptr)
107107
self.stream.sync()
108108

109-
# Return the result buffer as a CuPy array
109+
# Return the result as a NumPy array (on host).
110110
return cp.asnumpy(self.buffer).reshape(self.height, self.width, 4)
111111

112112

0 commit comments

Comments
 (0)