torch.backends
若API“是否支持”为“是”，“限制与说明”为空，说明此API和原生API支持度保持一致。
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.backends.cpu.get_cpu_capability |
是 |
|
torch.backends.cuda.is_built |
是 |
支持fp32 |
torch.backends.cuda.matmul.allow_tf32 |
是 |
支持fp32 |
torch.backends.cuda.preferred_linalg_library |
是 |
支持fp32 |
torch.backends.cuda.flash_sdp_enabled |
是 |
支持fp32 |
torch.backends.cuda.enable_mem_efficient_sdp |
是 |
支持fp32 |
torch.backends.cuda.mem_efficient_sdp_enabled |
是 |
支持fp32 |
torch.backends.cuda.enable_flash_sdp |
是 |
|
torch.backends.cuda.math_sdp_enabled |
是 |
支持fp32 |
torch.backends.cuda.enable_math_sdp |
是 |
支持fp32 |
torch.backends.cuda.sdp_kernel |
是 |
支持fp32 |
torch.backends.cudnn.is_available |
是 |
支持fp32 |
torch.backends.cudnn.enabled |
是 |
支持fp32 |
torch.backends.cudnn.allow_tf32 |
是 |
支持fp32 |
torch.backends.cudnn.deterministic |
是 |
|
torch.backends.cudnn.benchmark |
是 |
支持fp32 |
torch.backends.cudnn.benchmark_limit |
否 |
|
torch.backends.mps.is_available |
是 |
|
torch.backends.mps.is_built |
是 |
|
torch.backends.mkl.is_available |
是 |
|
torch.backends.mkl.verbose |
是 |
|
torch.backends.mkldnn.is_available |
Atlas 训练系列产品：是 Atlas A2 训练系列产品：是 |
|
torch.backends.mkldnn.verbose |
是 |
|
torch.backends.openmp.is_available |
是 |
支持fp32 |
torch.backends.opt_einsum.is_available |
是 |
支持fp32 |
torch.backends.opt_einsum.get_opt_einsum |
是 |
支持fp32 |
torch.backends.opt_einsum.enabled |
是 |
|
torch.backends.opt_einsum.strategy |
是 |