From 7f15d845a4df3dac7898517672c42cffc8c77b8c Mon Sep 17 00:00:00 2001 From: HydrogenSulfate <490868991@qq.com> Date: Mon, 16 Dec 2024 16:11:23 +0800 Subject: [PATCH 1/2] Update _backends.py "use paddle.is_tensor instead of isinstance to support pir.Value type" --- einops/_backends.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/einops/_backends.py b/einops/_backends.py index db43cb17..6bd2012a 100644 --- a/einops/_backends.py +++ b/einops/_backends.py @@ -623,7 +623,7 @@ def __init__(self): self.paddle = paddle def is_appropriate_type(self, tensor): - return isinstance(tensor, (self.paddle.Tensor, self.paddle.static.Variable)) + return paddle.is_tensor(tensor) def from_numpy(self, x): tensor = self.paddle.to_tensor(x) From cb8af1eaa98a3abe94c157799f7ac1179029d2c7 Mon Sep 17 00:00:00 2001 From: HydrogenSulfate <490868991@qq.com> Date: Mon, 16 Dec 2024 20:18:58 +0800 Subject: [PATCH 2/2] Update _backends.py --- einops/_backends.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/einops/_backends.py b/einops/_backends.py index 6bd2012a..b18a60b0 100644 --- a/einops/_backends.py +++ b/einops/_backends.py @@ -623,7 +623,7 @@ def __init__(self): self.paddle = paddle def is_appropriate_type(self, tensor): - return paddle.is_tensor(tensor) + return self.paddle.is_tensor(tensor) def from_numpy(self, x): tensor = self.paddle.to_tensor(x)