Skip to content

Commit 2c63250

Browse files
committed
minor fix
1 parent 0d4a5ed commit 2c63250

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

graph_net/paddle/test_compiler.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -82,9 +82,7 @@ def get_input_spec(args):
8282

8383

8484
def regular_item(item):
85-
if isinstance(item, paddle.Tensor) and (
86-
item.dtype == paddle.bfloat16 or item.dtype == paddle.bfloat32
87-
):
85+
if isinstance(item, paddle.Tensor) and (item.dtype == paddle.bfloat16):
8886
item = np.array(item.astype("float32"))
8987
else:
9088
item = np.array(item)

0 commit comments

Comments
 (0)