Skip to content

Commit 1bee724

Browse files
[Paddle] Update _backends.py to support pir.Value (#353)
* Update _backends.py "use paddle.is_tensor instead of isinstance to support pir.Value type" * Update _backends.py
1 parent 5906eb8 commit 1bee724

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

einops/_backends.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -551,7 +551,7 @@ def __init__(self):
551551
self.paddle = paddle
552552

553553
def is_appropriate_type(self, tensor):
554-
return isinstance(tensor, (self.paddle.Tensor, self.paddle.static.Variable))
554+
return self.paddle.is_tensor(tensor)
555555

556556
def from_numpy(self, x):
557557
tensor = self.paddle.to_tensor(x)

0 commit comments

Comments
 (0)