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

AutogradCPU

AutogradCUDA

AutogradVulkan

AutocastCPU

AutocastCUDA

AutocastVulkan

VmapCPU

VmapCUDA

VmapVulkan

Composite

VmapMode

DynamicLayerFrontMode

DynamicLayerBackMode

DispatchKey.name -> str