Skip to content

Commit 6a084dd

Browse files
committed
add note
1 parent 7d30b1f commit 6a084dd

File tree

1 file changed

+2
-0
lines changed
  • onnxscript/function_libs/torch_lib/ops

1 file changed

+2
-0
lines changed

onnxscript/function_libs/torch_lib/ops/core.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8536,6 +8536,8 @@ def aten_trunc(self: TFloat) -> TFloat:
85368536
@torch_op("math::trunc", trace_only=True)
85378537
def python_math_trunc(self: TFloat) -> TInt:
85388538
"""trunc(Tensor self) -> Tensor"""
8539+
# NOTE: This is used in SymInt/SymBool/SymFloat context, so
8540+
# we don't expect overflow to happen here.
85398541
return op.Cast(self, to=INT64.dtype)
85408542

85418543

0 commit comments

Comments
 (0)