legate.core.task.PyTask.cpu_variant

PyTask.cpu_variant(self, func: UserFunction)

Register a CPU variant for this task

Parameters:

func (UserFunction) – The new CPU variant function to call for all CPU executions.

Returns:

func unmodified.

Return type:

UserFunction

Raises:

RuntimeError – If the task has already completed registration.

Notes

Calling this method on a task with an existing CPU variant replaces the old variant with func. Therefore, this method may be used to update variants as well as add new ones.