The nvmath.bindings module is experimental and potentially subject to future changes.

xt_set_gpus

nvmath.bindings.cufft.xt_set_gpus(int handle, int n_gpus, which_gpus)

See cufftXtSetGPUs.