[DrRyanHuang]

PR Category

Execute Infrastructure

PR Types

Bug fixes

Description

torch 的 cumsum 函数，dtype 不为 None 时，会指定输出类型

import torch

x_int32 = torch.randint(1, 20, (10,)).int()
y = torch.cumsum(x_int32, dim=0, dtype=torch.int)
print(y.dtype) # torch.int32

#74625 引入后，导致 dtype 非 None 时，只要输入是 int （及其衍生类型），便会自动转化为 int64，显然不合理
导致 FastDeploy 运行 ERNIE 4.5 Turbo 系列模型全部挂掉

Paddle/python/paddle/tensor/math.py

Lines 4226 to 4234 in 955e32c

	if dtype is not None and x.dtype != convert_np_dtype_to_dtype_(dtype):
	x = cast(x, dtype)
	elif isinstance(x, paddle.Tensor) and x.dtype in [
	paddle.uint8,
	paddle.int8,
	paddle.int16,
	paddle.int32,
	]:
	x = cast(x, "int64")

Paddle 测试代码：

import paddle
from paddle.framework import convert_np_dtype_to_dtype_

seq_lens_this_time = paddle.randint(1, 128, (3,), dtype="int32")
cum_offsets_now = paddle.cumsum(128 - seq_lens_this_time, dtype='int32')

int32_type = convert_np_dtype_to_dtype_("int32")
assert cum_offsets_now.dtype == int32_type

[paddle-bot]

你的PR提交成功，感谢你对开源项目的贡献!
请关注后续CI自动化测试结果，详情请参考Paddle-CI手册。
Your PR has been submitted. Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

[SigureMo]

加一下单测

[codecov-commenter]

Codecov Report

✅ All modified and coverable lines are covered by tests.
⚠️ Please upload report for BASE (develop@2f156c7). Learn more about missing BASE report.

Additional details and impacted files

@@             Coverage Diff             @@
##             develop    #74830   +/-   ##
===========================================
  Coverage           ?   100.00%           
===========================================
  Files              ?         1           
  Lines              ?         6           
  Branches           ?         0           
===========================================
  Hits               ?         6           
  Misses             ?         0           
  Partials           ?         0           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

[SigureMo]

LGTMeow