torch.list_kernels

function list_kernels(options?: ListKernelsOptions): KernelInfo[]

List all registered kernels.

Parameters

optionsListKernelsOptionsoptional: Filter options

Returns

KernelInfo[]– Array of kernel information

Examples

// List all kernels
torch.library.list_kernels();

// List only WebGPU kernels
torch.library.list_kernels({ device: 'webgpu' });

// List kernels for a specific operation
torch.library.list_kernels({ op: 'add' });