

"torch.linalg.svd()"


"torch.linalg.svdvals()"


Return type:
      _LinalgBackend
class torch.backends.cuda.SDPBackend(value)
Enum class for the scaled dot product attention backends.
Warning:
 This flag is experimental and subject to change.'

This class needs to stay inline with the enum defined in:
   pytorch/aten/src/ATen/native/transformers/sdp_utils_cpp.h
torch.backends.cuda.flash_sdp_enabled()
Warning:
 This flag is experimental and subject to change.

Returns whether flash sdp is enabled or not.
torch.backends.cuda.enable_mem_efficient_sdp(enabled)
Warning:
 This flag is experimental and subject to change.

Enables or disables memory efficient sdp.
torch.backends.cuda.mem_efficient_sdp_enabled()
Warning:
 This flag is experimental and subject to change.

Returns whether memory efficient sdp is enabled or not.
torch.backends.cuda.enable_flash_sdp(enabled)
Warning:
 This flag is experimental and subject to change.
