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_init

shmem_module_init(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])

Initializes the device state in the compiled function module so that it's able to perform SHMEM operations. Must have completed device initialization prior to calling this function.

For NVSHMEM, this calls nvshmemx_cumodule_init(). For ROCSHMEM, this calls rocshmemx_hipmodule_init().

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 initialize with SHMEM.

Raises:

String: If module initialization 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