[WIP] Support JITFunction in preload #8794
Conversation
| if 'jit_function' in value: | ||
| jf_key = value['jit_function'] | ||
| if jf_key in _triton_jit_function_registry: | ||
| return _triton_jit_function_registry[jf_key] |
There was a problem hiding this comment.
preload is usually used in a different python process than the one where the specialization info was created, so the availability of the functions will depend on all imports happening before you run the preload calls.
Here is a reference solution that I wrote several years ago that does the import dynamically:
https://github.com/Quansight-Labs/uarray/blob/92216042c8e350cb174e6ddabbe70bf31132cac4/src/uarray/_backend.py#L64-L110
This removes the import dependence, but does limit it to only working on importable function; i.e. not defined in main or inside a function.
There was a problem hiding this comment.
thanks, yes that's what I'm trying to evaluate right now. Sounds like doing something like that would be a good direction
2 participants
No description provided.