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

AutogradCPU

AutogradCUDA

AutogradVulkan

AutogradSparseCPU

AutogradSparseCUDA

AutogradSparse

AutocastCPU

AutocastCUDA

AutocastVulkan

AutocastSparseCPU

AutocastSparseCUDA

AutocastSparse

VmapCPU

VmapCUDA

VmapVulkan

VmapSparseCPU

VmapSparseCUDA

VmapSparse

Composite

VmapMode

DynamicLayerFrontMode

DynamicLayerBackMode

Python

DispatchKey.name -> str