|
27 | 27 |
|
28 | 28 | _KernelCompileResult = namedtuple(
|
29 | 29 | "_KernelCompileResult",
|
30 |
| - ["status", "cres_or_error", "kernel_module"], |
| 30 | + ["status", "cres_or_error", "entry_point"], |
31 | 31 | )
|
32 | 32 |
|
33 | 33 |
|
@@ -155,7 +155,7 @@ def _compile_cached(
|
155 | 155 | "w",
|
156 | 156 | encoding="UTF-8",
|
157 | 157 | ) as f:
|
158 |
| - f.write(kernel_cres.library._final_module.__str__()) |
| 158 | + f.write(kernel_cres.library._final_module) |
159 | 159 |
|
160 | 160 | except errors.TypingError as e:
|
161 | 161 | self._failed_cache[key] = e
|
@@ -246,9 +246,9 @@ def typeof_pyval(self, val):
|
246 | 246 | self._types_active_call.append(tp)
|
247 | 247 | return tp
|
248 | 248 |
|
249 |
| - def add_bitcode_overload(self, cres, kernel_module): |
| 249 | + def add_overload(self, cres): |
250 | 250 | args = tuple(cres.signature.args)
|
251 |
| - self.overloads[args] = kernel_module |
| 251 | + self.overloads[args] = cres.entry_point |
252 | 252 |
|
253 | 253 | def compile(self, sig) -> _KernelCompileResult:
|
254 | 254 | disp = self._get_dispatcher_for_current_target()
|
@@ -311,13 +311,11 @@ def folded(args, kws):
|
311 | 311 | )[1]
|
312 | 312 |
|
313 | 313 | raise e.bind_fold_arguments(folded)
|
314 |
| - self.add_bitcode_overload( |
315 |
| - kcres.cres_or_error, kcres.kernel_module |
316 |
| - ) |
| 314 | + self.add_overload(kcres.cres_or_error) |
317 | 315 |
|
318 |
| - # FIXME: enable caching |
| 316 | + # TODO: enable caching of kernel_module |
319 | 317 |
|
320 |
| - return kcres.kernel_module |
| 318 | + return kcres.entry_point |
321 | 319 |
|
322 | 320 | def __getitem__(self, args):
|
323 | 321 | """Square-bracket notation for configuring launch arguments is not
|
|
0 commit comments