From 92a12b188a0f2d819665ffbfd0516dd9ec1fcb8b Mon Sep 17 00:00:00 2001 From: Weida Hong Date: Fri, 20 Feb 2026 13:02:50 +0000 Subject: [PATCH] Upgrade usage of deprecated PyTorch API register_fake is favored over impl_abstract since PyTorch 2.4. Upgrade the API usage to resolve FutureWarning. Signed-off-by: Weida Hong --- test/test_libraries.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_libraries.py b/test/test_libraries.py index cbac6a6..a527f72 100644 --- a/test/test_libraries.py +++ b/test/test_libraries.py @@ -16,7 +16,7 @@ import torch import torch.nn.functional as F -from torch.library import Library, impl, impl_abstract +from torch.library import Library, impl, register_fake import torchax import torchax.export @@ -44,7 +44,7 @@ def _mylib_scaled_dot_product_attention(q, k, v): return y.transpose(1, 2) -@impl_abstract("mylib::scaled_dot_product_attention") +@register_fake("mylib::scaled_dot_product_attention") def _mylib_scaled_dot_product_attention_meta(q, k, v): return torch.empty_like(q)