mscclpp.utils

Functions

get_device_arch()
is_torch_tensor(tensor)
pack(*args)
torch_dtype_to_mscclpp_dtype(dtype)

Classes

Kernel(cubin, kernel_name)
KernelBuilder(file, kernel_name[, file_dir, ...])