For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).

Mojo function

shmem_module_finalize

shmem_module_finalize(device_function: DeviceFunction[target=device_function.target, compile_options=device_function.compile_options, link_options=device_function.link_options, _ptxas_info_verbose=device_function._ptxas_info_verbose])

Finalizes the device state in the compiled function module and cleans up NVSHMEM operations. This should be called when NVSHMEM operations are no longer needed for the given device function.

Args:

• ​device_function (DeviceFunction[target=device_function.target, compile_options=device_function.compile_options, link_options=device_function.link_options, _ptxas_info_verbose=device_function._ptxas_info_verbose]): The compiled device function to finalize and clean up NVSHMEM resources.

Raises:

String: If module finalization fails.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit