Skip to content

Commit 742792f

Browse files
add floor and ceil with same return dtype
1 parent 603c852 commit 742792f

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

array_api_compat/paddle/_aliases.py

+9-1
Original file line numberDiff line numberDiff line change
@@ -1036,7 +1036,7 @@ def is_complex(dtype):
10361036
paddle.framework.core.DataType.FLOAT64,
10371037
paddle.framework.core.DataType.FLOAT16,
10381038
paddle.framework.core.DataType.BFLOAT16,
1039-
]
1039+
]
10401040
elif kind == "complex floating":
10411041
return is_complex(dtype)
10421042
elif kind == "numeric":
@@ -1137,6 +1137,14 @@ def asarray(
11371137
return obj
11381138

11391139

1140+
def floor(x: array, /) -> array:
1141+
return paddle.floor(x).to(x.dtype)
1142+
1143+
1144+
def ceil(x: array, /) -> array:
1145+
return paddle.ceil(x).to(x.dtype)
1146+
1147+
11401148
def clip(
11411149
x: array,
11421150
/,

0 commit comments

Comments
 (0)