mscclpp.utils

Functions

is_torch_tensor(tensor)
pack(*args)

Classes

GpuBuffer(*args, **kwargs)
Kernel(ptx, kernel_name)
KernelBuilder(file, kernel_name[, file_dir, ...])

class mscclpp.utils.GpuBuffer(*args, **kwargs)

Bases: ndarray

Parameters:

• shape (int | Tuple[int])

• dtype (cupy.dtype)

• strides (Tuple[int])

• order (str)