torch.BaseKernelConfig
export interface BaseKernelConfig<Op extends OpName, DT extends DType, Dev extends Device> {
readonly op: Op;
readonly dtype: DT;
readonly device: Dev;
/**
* Optional predicate to conditionally use this kernel.
* Useful for specialized implementations (e.g., Strassen for large matrices).
*/
readonly predicate?: KernelPredicate;
/**
* Whether this kernel can be fused with other pointwise operations.
*/
readonly fusable?: boolean;
}Opextends OpNameDTextends DTypeDevextends Device- readonly
op(Op) - readonly
dtype(DT) - readonly
device(Dev) - readonly
predicate(KernelPredicate)optional - – Optional predicate to conditionally use this kernel. Useful for specialized implementations (e.g., Strassen for large matrices).
- readonly
fusable(boolean)optional - – Whether this kernel can be fused with other pointwise operations.
Base kernel configuration options.