DispatchKey

class tensorplay.DispatchKey

Dispatch keys of the runtime, in dispatch-priority order: autocast keys outrank autograd keys, which outrank backend keys.

Members:

CPU

CUDA

Vulkan

SparseCPU

SparseCUDA

Sparse

Meta

MPS

AutogradCPU

AutogradCUDA

AutogradVulkan

AutogradSparseCPU

AutogradSparseCUDA

AutogradSparse

AutogradMeta

AutogradMPS

AutocastCPU

AutocastCUDA

AutocastVulkan

AutocastSparseCPU

AutocastSparseCUDA

AutocastSparse

AutocastMeta

AutocastMPS

VmapCPU

VmapCUDA

VmapVulkan

VmapSparseCPU

VmapSparseCUDA

VmapSparse

VmapMeta

VmapMPS

Composite

VmapMode

DynamicLayerFrontMode

DynamicLayerBackMode

Python

DispatchKey.name -> str