fix: 修复 FFT 算子 host 侧缓冲区尺寸计算的整数溢出-ops-fft-AtomGit

已合并

fix: 修复 FFT 算子 host 侧缓冲区尺寸计算的整数溢出 #33

Tian_1122创建于 12 天前从Tian_1122/ops-fft： fix/fft-integer-overflow 合入到 cann/ops-fft： master

fix: 修复 FFT 算子 host 侧缓冲区尺寸计算的整数溢出 #33

已合并

Tian_1122创建于 12 天前

从Tian_1122/ops-fft fix/fft-integer-overflow合入到cann/ops-fft master

Tian_1122成员

12 天前

Pull Request已成功合入, 合并人@CANN-robot

（感谢 Tian_1122 的贡献）

Tian_1122成员

12 天前创建了 pull request，commit 9c6f61eb

atomgit-bot

12 天前评论：

变更摘要

本 PR 主要修复 FFT 系列算子 host 侧缓冲区尺寸计算中的整数溢出问题。原先 inputSize、outputSize、workspaceSize 等尺寸变量使用 uint32_t 保存，且在 n * batches * sizeof(float) * 2 这类连乘中先按 32 位整型计算，导致较大尺寸下发生溢出。此次改动将相关尺寸变量统一改为 size_t（个别改为 uint64_t），并在首个乘数上通过 static_cast<size_t>(...) 或 static_cast<uint64_t>(...) 强制提升为 64 位运算，从而保证缓冲区尺寸计算正确。

主要改动

一维 FFT 基础/混合/N 点算子尺寸计算提升为 64 位: 在 fft1_d_b.cpp、fft1_d_mix.cpp、fft1_d_n.cpp 中，将 inputSize、outputSize 由 uint32_t 改为 size_t，并用 static_cast<size_t>(n) * batches * sizeof(float) * 2 避免中间结果溢出。
带 stride 的一维 FFT 算子扩大 workspace 尺寸类型: 在 fft1_d_stride.cpp 中，除 inputSize、outputSize 外，还将 kernelWorkspaceSize 与 totalWorkspaceSize 改为 size_t，并对 n 使用 static_cast<size_t>(...)。
二维 FFT 算子 workspace 尺寸改为 uint64_t: 在 fft2_d_dd.cpp 中，inputSize、outputSize 改为 size_t，workspaceSize 改为 uint64_t 并通过 static_cast<uint64_t>(batches) 强制 64 位计算。
实数/逆实数 FFT 算子缓冲区尺寸改为 size_t: 在 rfft1_d_r2c_fft.cpp、irfft1_d_c2r_fft.cpp、irfft1_d_fft.cpp 中，将 inputSize、outputSize 及 workspaceSize 改为 size_t，并对 n 或 batches 使用 static_cast<size_t>(...) 防止 32 位溢出。

atomgit-bot

12 天前评论：

代码审查

✅ 未发现问题

CANN-robot成员

12 天前添加了label：cann-cla/yes

CANN-robot成员

12 天前评论：

CLA Signature Pass

Tian_1122, thanks for your pull request. All authors of the commits have signed the CLA. 👍

CANN-robot成员

12 天前评论：

Thanks for your pull-request.
The full list of commands accepted by me can be found at here.
You can get sig-info at here.
For more, you also can visit HICANN.

PR Approval Progress

✅ Congratulations! All modules have met the lgtm and approve requirements.

Module Approval Details

module	lgtm status	approve status
repo-cann/ops-fft	✅ 张浩, east_yang (2/2)	✅ 张浩, east_yang (2/1)

💡 Tip:

Committer can comment /approve or /lgtm

Commenting /approve implies both code review (lgtm) and intent to merge (approve)

Tian_1122成员

12 天前修改了pull request 的描述

此处折叠了19条事件消息查看更多

CANN-robot成员

12 天前添加了label：cann-cla/yes

Tian_1122成员

12 天前评论：

compile

CANN-robot成员

12 天前添加了label：ci-pipeline-running

CANN-robot成员

12 天前评论：

流水线任务触发成功
任务链接 [878a2497ee164fb2aecc850e7bc9fdb7][流水线指导]

任务名称	状态	日志	下载链接
Compile_Ascend_X86	✅ SUCCESS	>>>>>	>>>>>
Compile_Ascend_ARM	✅ SUCCESS	>>>>>	>>>>>
codecheck	✅ SUCCESS	>>>>>
SCA	✅ SUCCESS	>>>>>
antipoison	✅ SUCCESS	>>>>>
Check_Pr	✅ SUCCESS	>>>>>
pre_comment	✅ SUCCESS	>>>>>
Compile_Ascend_X86_ubuntu24	✅ SUCCESS	>>>>>	>>>>>
Compile_Ascend_ARM_ubuntu24	✅ SUCCESS	>>>>>	>>>>>
precommit	✅ SUCCESS	>>>>>
codecheck_codestyle	✅ SUCCESS	>>>>>