`torch.backends.cuda.sdp_kernel()` is deprecated. use `torch.nn.attention.sdpa_kernel()`

fpgaminer / joytag

The JoyTag Image Tagging Model

Apache License 2.0

400 stars 25 forks source link

`torch.backends.cuda.sdp_kernel()` is deprecated. use `torch.nn.attention.sdpa_kernel()` #11

Open MoConWu opened 3 months ago

MoConWu commented 3 months ago

pytouch feature： torch.backends.cuda.sdp_kernel() is deprecated. In the future, this context manager will be removed. Please see torch.nn.attention.sdpa_kernel() for the new context manager, with updated signature.’