torch.backends
若API“是否支持”为“是”，“限制与说明”为“-”，说明此API和原生API支持度保持一致。
API名称 |
NPU形式名称 |
是否支持 |
限制与说明 |
|---|---|---|---|
torch.backends.cpu.get_cpu_capability |
- |
是 |
- |
torch.backends.cuda.is_built |
- |
是 |
支持fp32 |
torch.backends.cuda.matmul.allow_tf32 |
- |
是 |
支持fp32 |
torch.backends.cuda.cufft_plan_cache |
torch.npu.backends.fft_plan_cache |
是 |
- |
torch.backends.cuda.cufft_plan_cache.size |
torch.npu.backends.fft_plan_cache.size |
是 |
- |
torch.backends.cuda.cufft_plan_cache.max_size |
torch.npu.backends.fft_plan_cache.max_size |
是 |
输入范围为1-99 |
torch.backends.cuda.cufft_plan_cache.clear |
torch.npu.backends.fft_plan_cache.clear() |
是 |
- |
torch.backends.cuda.preferred_linalg_library |
- |
否 |
- |
torch.backends.cuda.flash_sdp_enabled |
- |
否 |
- |
torch.backends.cuda.enable_mem_efficient_sdp |
- |
否 |
- |
torch.backends.cuda.mem_efficient_sdp_enabled |
- |
否 |
- |
torch.backends.cuda.enable_flash_sdp |
- |
否 |
- |
torch.backends.cuda.math_sdp_enabled |
- |
否 |
- |
torch.backends.cuda.enable_math_sdp |
- |
否 |
- |
torch.backends.cuda.sdp_kernel |
- |
否 |
- |
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 |
- |
- |
|
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 |
- |
否 |
- |