Dmitry Rogozhkin 31830474bf Fix test_eager_matches_sdpa_inference for XPU backend (#34889) ... * Use torch.nn.attention.sdpa_kernel instead of deprecated torch.backends.cuda.sdp_kernel Signed-off-by: Dmitry Rogozhkin <dmitry.v.rogozhkin@intel.com> * Fix test_eager_matches_sdpa_inference for XPU backend As of PyTorch 2.5 XPU backend supports only torch.nn.attention.SDPBackend.MATH which is implemented on PyTorch level using aten operators and is device agnostic with respect to implementation of each aten operator. Thus, we can reuse CUDA (or CPU) MATH weights for XPU. Fixes: #34888 Signed-off-by: Dmitry Rogozhkin <dmitry.v.rogozhkin@intel.com> * Use torch.amp.autocast instead of deprecated torch.cuda.amp.autocast in nemotron Signed-off-by: Dmitry Rogozhkin <dmitry.v.rogozhkin@intel.com> --------- Signed-off-by: Dmitry Rogozhkin <dmitry.v.rogozhkin@intel.com>		2024-12-02 16:21:04 +01:00
..
__init__.py	Add Musicgen (#24109)	2023-06-29 14:48:59 +01:00
test_modeling_musicgen.py	Fix test_eager_matches_sdpa_inference for XPU backend (#34889)	2024-12-02 16:21:04 +01:00
test_processor_musicgen.py	rename all test_processing_*.py to test_processor_*.py (#33878)	2024-10-02 16:43:43 +02:00