Skip to main content

Module offload

Module offload 

Source

Functionsยง

compile_for_device ๐Ÿ”’
expand_kernel ๐Ÿ”’
The offload_kernel macro expands the function into two separate definitions: one on the host to handle the call, and one on the device for executing the kernel.
extract_fn ๐Ÿ”’
outer_normal_attr ๐Ÿ”’