Enums§
- QEngine
- Quantization engines
Functions§
- get_
num_ interop_ threads - Get the number of threads used by torch for inter-op parallelism.
- get_
num_ threads - Get the number of threads used by torch in parallel regions.
- has_
cuda - has_
cudart - has_
cusolver - has_hip
- has_ipu
- has_
lapack - has_
lazy - has_
magma - has_mkl
- has_
mkldnn - has_mps
- has_
openmp - has_
vulkan - Check whether the vulkan backend is available. None that this backend is not included by default as of PyTorch 2.0.0. https://pytorch.org/tutorials/prototype/vulkan_workflow.html#building-pytorch-with-vulkan-backend
- has_xla
- manual_
seed - Sets the random seed used by torch.
- set_
num_ interop_ threads - Set the number of threads used by torch for inter-op parallelism.
- set_
num_ threads - Set the number of threads used by torch in parallel regions.
- version_
cudart - version_
cudnn